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