| 175 | return len(self.X) |
| 176 | |
| 177 | class DataHandler2(Dataset): |
| 178 | def __init__(self, X, Y, transform=None): |
| 179 | self.X = X |
| 180 | self.Y = Y |
| 181 | self.transform = transform |
| 182 | |
| 183 | def __getitem__(self, index): |
| 184 | x, y = self.X[index], self.Y[index] |
| 185 | if self.transform is not None: |
| 186 | x = Image.fromarray(np.transpose(x, (1, 2, 0))) |
| 187 | x = self.transform(x) |
| 188 | return x, y, index |
| 189 | |
| 190 | def __len__(self): |
| 191 | return len(self.X) |
| 192 | |
| 193 | class DataHandler3(Dataset): |
| 194 | def __init__(self, X, Y, transform=None): |
nothing calls this directly
no outgoing calls
no test coverage detected