new reversed convnext

2023-01-09 21:22:15 +00:00
parent 93357f81cb
commit bca2e0bd2c
3 changed files with 161 additions and 35 deletions
@@ -190,10 +190,8 @@ class Dataset(Dataset):
        # v2i_coverage[v2i[:,0], v2i[:, 1]] = 1
        # ex.imshow(v2i_coverage).show()

-
        rgb_image = tv.io.read_image(os.path.join(self.folder_rgb, frame.id, f'{frame.frame_nr:04d}.png'), mode=tv.io.ImageReadMode.RGB).to(self.device).to(torch.float32)

-
        # image = self.transforms(image)
        data = { 'mesh_posed': mesh_posed,
                 'depth_image': depth_image,