| 86 | |
| 87 | |
| 88 | class IndexedDataset(Dataset): |
| 89 | def __init__(self, dataset): |
| 90 | assert isinstance(dataset, Dataset) |
| 91 | self.loader = dataset.loader |
| 92 | self.classes = dataset.classes |
| 93 | self.samples = dataset.samples |
| 94 | self.y = dataset.y |
| 95 | self.transform = dataset.transform |
| 96 | self.target_transform = None |
| 97 | |
| 98 | def __getitem__(self, idx): |
| 99 | x, y = super().__getitem__(idx) |
| 100 | return x, y, idx |
| 101 | |
| 102 | def __len__(self): |
| 103 | return len(self.y) |
| 104 | |
| 105 | |
| 106 | class Loader(): |
no outgoing calls
no test coverage detected