bug fix

mesh_reconstruction/recon.py

@@ -12,7 +12,7 @@ from scripts.utils import to_py3d_mesh, init_target
 def reconstruct_stage1(pils: List[Image.Image], steps=100, vertices=None, faces=None, start_edge_len=0.15, end_edge_len=0.005, decay=0.995, return_mesh=True, loss_expansion_weight=0.1, gain=0.1):
     vertices, faces = vertices.to("cuda"), faces.to("cuda")
     assert len(pils) == 4
-    mv,proj = make_star_cameras_orthographic_py3d([0, 270, 180, 90], device="cuda", focal=1., dist=4.0)
+    cameras = make_star_cameras_orthographic_py3d([0, 270, 180, 90], device="cuda", focal=1., dist=4.0)
     renderer = Pytorch3DNormalsRenderer(cameras, list(pils[0].size), device="cuda")
     
     target_images = init_target(pils, new_bkgd=(0., 0., 0.)) # 4s

mesh_reconstruction/refine.py

@@ -17,7 +17,7 @@ def run_mesh_refine(vertices, faces, pils: List[Image.Image], steps=100, start_e
     poission_steps = []
 
     assert len(pils) == 4
-    mv,proj = make_star_cameras_orthographic_py3d([0, 270, 180, 90], device="cuda", focal=1., dist=4.0)     
+    cameras = make_star_cameras_orthographic_py3d([0, 270, 180, 90], device="cuda", focal=1., dist=4.0)     
     renderer = Pytorch3DNormalsRenderer(cameras, list(pils[0].size), device="cuda")
 
     target_images = init_target(pils, new_bkgd=(0., 0., 0.)) # 4s