| 55 | |
| 56 | |
| 57 | class IndexedDataset(): |
| 58 | def __init__(self, x, y, transform=None): |
| 59 | self.x = x |
| 60 | self.y = y |
| 61 | self.ii = np.array( range(len(x)), dtype=np.int64 ) |
| 62 | self.transform = transform |
| 63 | |
| 64 | def __getitem__(self, idx): |
| 65 | x, y, ii = Image.fromarray(self.x[idx]), self.y[idx], self.ii[idx] |
| 66 | if self.transform is not None: |
| 67 | x = self.transform(x) |
| 68 | return x, y, ii |
| 69 | |
| 70 | def __len__(self): |
| 71 | return len(self.x) |
| 72 | |
| 73 | |
| 74 | def datasetCIFAR10(root='./path', train=True, transform=None): |
nothing calls this directly
no outgoing calls
no test coverage detected