(self)
| 732 | self.index = 0 |
| 733 | |
| 734 | def load_data(self): |
| 735 | image = self.rgb_loader(self.images[self.index]) |
| 736 | image = self.transform(image).unsqueeze(0) |
| 737 | |
| 738 | gt = self.binary_loader(self.gts[self.index]) |
| 739 | # edge = self.binary_loader(self.edges[self.index]) |
| 740 | |
| 741 | name = self.images[self.index].split('/')[-1] |
| 742 | |
| 743 | image_for_post = self.rgb_loader(self.images[self.index]) |
| 744 | image_for_post = image_for_post.resize(gt.size) |
| 745 | |
| 746 | if name.endswith('.jpg'): |
| 747 | name = name.split('.jpg')[0] + '.png' |
| 748 | |
| 749 | self.index += 1 |
| 750 | self.index = self.index % self.size |
| 751 | |
| 752 | return image, gt, name, np.array(image_for_post) |
| 753 | |
| 754 | def rgb_loader(self, path): |
| 755 | with open(path, 'rb') as f: |
no test coverage detected