(self, index)
| 52 | |
| 53 | |
| 54 | def __getitem__(self, index): |
| 55 | |
| 56 | image = self.rgb_loader(self.images[index]) |
| 57 | gt = self.binary_loader(self.gts[index]) |
| 58 | |
| 59 | seed = np.random.randint(2147483647) # make a seed with numpy generator |
| 60 | random.seed(seed) # apply this seed to img tranfsorms |
| 61 | torch.manual_seed(seed) # needed for torchvision 0.7 |
| 62 | if self.img_transform is not None: |
| 63 | image = self.img_transform(image) |
| 64 | |
| 65 | random.seed(seed) # apply this seed to img tranfsorms |
| 66 | torch.manual_seed(seed) # needed for torchvision 0.7 |
| 67 | if self.gt_transform is not None: |
| 68 | gt = self.gt_transform(gt) |
| 69 | return image, gt |
| 70 | |
| 71 | def filter_files(self): |
| 72 | assert len(self.images) == len(self.gts) |
nothing calls this directly
no test coverage detected