| 62 | |
| 63 | |
| 64 | class SubDataset: |
| 65 | def __init__(self, sub_meta, cl, transform=transforms.ToTensor(), target_transform=identity): |
| 66 | self.sub_meta = sub_meta |
| 67 | self.cl = cl |
| 68 | self.transform = transform |
| 69 | self.target_transform = target_transform |
| 70 | |
| 71 | def __getitem__(self, i): |
| 72 | # print( '%d -%d' %(self.cl,i)) |
| 73 | image_path = os.path.join(self.sub_meta[i]) |
| 74 | img = Image.open(image_path).convert('RGB') |
| 75 | img = self.transform(img) |
| 76 | target = self.target_transform(self.cl) |
| 77 | return img, target |
| 78 | |
| 79 | def __len__(self): |
| 80 | return len(self.sub_meta) |
| 81 | |
| 82 | |
| 83 | class EpisodicBatchSampler(object): |