Integrate CharacterGen & Improve its workflow with Unique3D

gen_3d_modules/Unique3D/mesh_reconstruction/recon.py

@@ -12,7 +12,6 @@ from .render import NormalsRenderer, Pytorch3DNormalsRenderer
 from Unique3D.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.1, end_edge_len=0.02, decay=0.995, return_mesh=False, loss_expansion_weight=0.1, gain=0.05):
-    vertices, faces = vertices.to("cuda"), faces.to("cuda")
     assert len(pils) == 4
     mv,proj = make_star_cameras_orthographic(4, 1)
     renderer = NormalsRenderer(mv,proj,list(pils[0].size))
@@ -24,7 +23,7 @@ def reconstruct_stage1(pils: List[Image.Image], steps=100, vertices=None, faces=
     target_images = target_images[[0, 3, 2, 1]]
 
     # 2. init from coarse mesh
-    opt = MeshOptimizer(vertices,faces, local_edgelen=False, gain=gain, edge_len_lims=(end_edge_len, start_edge_len))
+    opt = MeshOptimizer(vertices, faces, local_edgelen=False, gain=gain, edge_len_lims=(end_edge_len, start_edge_len))
 
     vertices = opt.vertices
 

gen_3d_modules/Unique3D/mesh_reconstruction/remesh.py

@@ -95,7 +95,6 @@ def calc_vertex_normals(
         faces:torch.Tensor, #F,3 long, first face may be all zero
         face_normals:torch.Tensor=None, #F,3, not normalized
         )->torch.Tensor: #F,3
-
     F = faces.shape[0]
 
     if face_normals is None: