Spaces:

CVPR
/

monoscene_lite

Runtime error

anhquancao commited on Jun 29, 2022

Commit

4d7dc2a

•

1 Parent(s): ded3155

Update monoscene/monoscene.py

Files changed (1) hide show

monoscene/monoscene.py CHANGED Viewed

@@ -96,13 +96,15 @@ class MonoScene(pl.LightningModule):
                 if x3d is None:
                     x3d = self.projects[str(scale_2d)](
                         x_rgb["1_" + str(scale_2d)][i],
-                        projected_pix // scale_2d,
                         fov_mask,
                     )
                 else:
                     x3d += self.projects[str(scale_2d)](
                         x_rgb["1_" + str(scale_2d)][i],
-                        projected_pix // scale_2d,
                         fov_mask,
                     )
             x3ds.append(x3d)

                 if x3d is None:
                     x3d = self.projects[str(scale_2d)](
                         x_rgb["1_" + str(scale_2d)][i],
+                        torch.div(projected_pix, scale_2d, rounding_mode='floor'),
+                        # projected_pix // scale_2d,
                         fov_mask,
                     )
                 else:
                     x3d += self.projects[str(scale_2d)](
                         x_rgb["1_" + str(scale_2d)][i],
+                        torch.div(projected_pix, scale_2d, rounding_mode='floor'),
+                        # projected_pix // scale_2d,
                         fov_mask,
                     )
             x3ds.append(x3d)