(self, sample)
| 39 | # self.resize = transforms.Resize((375, 1242)) |
| 40 | |
| 41 | def __call__(self, sample): |
| 42 | image, depth = sample['image'], sample['depth'] |
| 43 | |
| 44 | image = self.to_tensor(image) |
| 45 | image = self.normalize(image) |
| 46 | depth = self.to_tensor(depth) |
| 47 | |
| 48 | # image = self.resize(image) |
| 49 | |
| 50 | return {'image': image, 'depth': depth, 'dataset': "vkitti"} |
| 51 | |
| 52 | def to_tensor(self, pic): |
| 53 |