| 191 | return len(self.X) |
| 192 | |
| 193 | class DataHandler3(Dataset): |
| 194 | def __init__(self, X, Y, transform=None): |
| 195 | self.X = X |
| 196 | self.Y = Y |
| 197 | self.transform = transform |
| 198 | |
| 199 | def __getitem__(self, index): |
| 200 | x, y = self.X[index], self.Y[index] |
| 201 | if self.transform is not None: |
| 202 | x = Image.fromarray(x) |
| 203 | x = self.transform(x) |
| 204 | return x, y, index |
| 205 | |
| 206 | def __len__(self): |
| 207 | return len(self.X) |
| 208 | |
| 209 | class DataHandler4(Dataset): |
| 210 | def __init__(self, X, Y, transform=None): |
nothing calls this directly
no outgoing calls
no test coverage detected