(self)
| 153 | self.index = 0 |
| 154 | |
| 155 | def load_data(self): |
| 156 | image = self.rgb_loader(self.images[self.index]) |
| 157 | image = self.transform(image).unsqueeze(0) |
| 158 | gt = self.binary_loader(self.gts[self.index]) |
| 159 | name = self.images[self.index].split('/')[-1] |
| 160 | if name.endswith('.jpg'): |
| 161 | name = name.split('.jpg')[0] + '.png' |
| 162 | self.index += 1 |
| 163 | return image, gt, name |
| 164 | |
| 165 | def rgb_loader(self, path): |
| 166 | with open(path, 'rb') as f: |
nothing calls this directly
no test coverage detected