Minor precision fix on unproject

facebookresearch · Mar 11, 2023 · e0624ee · e0624ee
1 parent 013ff4f
commit e0624ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch3d/renderer/cameras.py b/pytorch3d/renderer/cameras.py
@@ -135,7 +135,7 @@ def unproject_points(self, xy_depth: torch.Tensor, **kwargs):
             xy_depth = torch.cat((xy, depth), dim=2)
             # unproject to the world coordinates
             xyz_unproj_world = cameras.unproject_points(xy_depth, world_coordinates=True)
-            print(torch.allclose(xyz, xyz_unproj_world)) # True
+            print(torch.allclose(xyz, xyz_unproj_world, atol=1e-4)) # True
             # unproject to the camera coordinates
             xyz_unproj = cameras.unproject_points(xy_depth, world_coordinates=False)
             print(torch.allclose(xyz_cam, xyz_unproj)) # True