| """ data sample | |
| """ | |
| class Sample(): | |
| def __init__(self, img=None, | |
| depth=None, depth_mask=None, | |
| normal=None, normal_mask=None, | |
| intrins=None, flipped=False, | |
| dataset_name='dataset', scene_name='scene', img_name='img', | |
| info={}): | |
| self.img = img # input image | |
| self.depth = depth # depth - GT | |
| self.depth_mask = depth_mask # depth - valid_mask | |
| self.normal = normal # surface normals - GT | |
| self.normal_mask = normal_mask # surface normals - valid_mask | |
| self.intrins = intrins # camera intrinsics | |
| self.flipped = flipped # True when the image is flipped during augmentation | |
| self.dataset_name = dataset_name | |
| self.scene_name = scene_name | |
| self.img_name = img_name | |
| # other info (this is a dict containing any additional information) | |
| self.info = info |