bug fix

mesh_reconstruction/func.py

@@ -115,4 +115,19 @@ def make_sphere(level:int=2,radius=1.,device='cuda') -> Tuple[torch.Tensor,torch
     sphere = trimesh.creation.icosphere(subdivisions=level, radius=1.0, color=None)
     vertices = torch.tensor(sphere.vertices, device=device, dtype=torch.float32) * radius
     faces = torch.tensor(sphere.faces, device=device, dtype=torch.long)
-    return vertices,faces
+    return vertices,faces
+
+from pytorch3d.renderer import (
+    FoVOrthographicCameras,
+    look_at_view_transform,
+)
+
+def get_camera(R, T, focal_length=1 / (2**0.5)):
+    focal_length = 1 / focal_length
+    camera = FoVOrthographicCameras(device=R.device, R=R, T=T, min_x=-focal_length, max_x=focal_length, min_y=-focal_length, max_y=focal_length)
+    return camera
+
+def make_star_cameras_orthographic_py3d(azim_list, device, focal=2/1.35, dist=1.1):
+    R, T = look_at_view_transform(dist, 0, azim_list)
+    focal_length = 1 / focal
+    return FoVOrthographicCameras(device=R.device, R=R, T=T, min_x=-focal_length, max_x=focal_length, min_y=-focal_length, max_y=focal_length).to(device)

mesh_reconstruction/render.py

@@ -123,16 +123,6 @@ class Pytorch3DNormalsRenderer:
         mesh = Meshes(verts=[vertices], faces=[faces], textures=TexturesVertex(verts_features=[(normals + 1) / 2])).to(self.device)
         return render_mesh_vertex_color(mesh, self.cameras, self._image_size[0], self._image_size[1], device=self.device)
     
-def get_camera(R, T, focal_length=1 / (2**0.5)):
-    focal_length = 1 / focal_length
-    camera = FoVOrthographicCameras(device=R.device, R=R, T=T, min_x=-focal_length, max_x=focal_length, min_y=-focal_length, max_y=focal_length)
-    return camera
-
-def make_star_cameras_orthographic_py3d(azim_list, device, focal=2/1.35, dist=1.1):
-    R, T = look_at_view_transform(dist, 0, azim_list)
-    focal_length = 1 / focal
-    return FoVOrthographicCameras(device=R.device, R=R, T=T, min_x=-focal_length, max_x=focal_length, min_y=-focal_length, max_y=focal_length).to(device)
-
 def save_tensor_to_img(tensor, save_dir):
     from PIL import Image
     import numpy as np
@@ -146,7 +136,7 @@ if __name__ == "__main__":
     import sys
     import os
     sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
-    from mesh_reconstruction.func import make_star_cameras_orthographic
+    from mesh_reconstruction.func import make_star_cameras_orthographic, make_star_cameras_orthographic_py3d
     cameras = make_star_cameras_orthographic_py3d([0, 270, 180, 90], device="cuda", focal=1., dist=4.0)
     mv,proj = make_star_cameras_orthographic(4, 1)
     resolution = 1024