(self, resize_shape)
| 33 | |
| 34 | class ToTensor(object): |
| 35 | def __init__(self, resize_shape): |
| 36 | # self.normalize = transforms.Normalize( |
| 37 | # mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) |
| 38 | self.normalize = lambda x : x |
| 39 | self.resize = transforms.Resize(resize_shape) |
| 40 | |
| 41 | def __call__(self, sample): |
| 42 | image, depth = sample['image'], sample['depth'] |
nothing calls this directly
no outgoing calls
no test coverage detected