debug

models/DiMeR/models/DiMeR.py

@@ -86,8 +86,11 @@ class DiMeR(nn.Module):
         
     @spaces.GPU
     def init_flexicubes_geometry(self, device, fovy=50.0):
+        print(1)
         camera = PerspectiveCamera(fovy=fovy, device=device)
+        print(2)
         renderer = NeuralRender(device, camera_model=camera)
+        print(3)
         self.geometry = FlexiCubesGeometry(
             grid_res=self.grid_res, 
             scale=self.grid_scale, 

models/DiMeR/models/geometry/camera/perspective_camera.py

@@ -24,9 +24,13 @@ def projection(x=0.1, n=1.0, f=50.0, near_plane=None):
 class PerspectiveCamera(Camera):
     def __init__(self, fovy=49.0, device='cuda'):
         super(PerspectiveCamera, self).__init__()
+        print(1.1)
         self.device = device
+        print(1.2)
         focal = np.tan(fovy / 180.0 * np.pi * 0.5)
+        print(1.3)
         self.proj_mtx = torch.from_numpy(projection(x=focal, f=1000.0, n=1.0, near_plane=0.1)).to(self.device).unsqueeze(dim=0)
+        print(1.4)
 
     def project(self, points_bxnx4):
         out = torch.matmul(

models/DiMeR/models/geometry/render/neural_render.py

@@ -71,10 +71,15 @@ def compute_vertex_normal(v_pos, t_pos_idx):
 class NeuralRender(Renderer):
     def __init__(self, device='cuda', camera_model=None):
         super(NeuralRender, self).__init__()
+        print(2.1)
         self.device = device
+        print(2.2)
         self.ctx = dr.RasterizeCudaContext(device=device)
+        print(2.3)
         self.projection_mtx = None
+        print(2.4)
         self.camera = camera_model
+        print(2.5)
         
     # ==============================================================================================
     #  pixel shader

models/DiMeR/models/geometry/rep_3d/flexicubes_geometry.py

@@ -40,26 +40,35 @@ class FlexiCubesGeometry(Geometry):
             self, grid_res=64, scale=2.0, device='cuda', renderer=None,
             render_type='neural_render', args=None):
         super(FlexiCubesGeometry, self).__init__()
+        print(3.1)
         self.grid_res = grid_res
         self.device = device
         self.args = args
+        print(3.2)
         self.fc = FlexiCubes(device, weight_scale=0.5)
+        print(3.3)
         self.verts, self.indices = self.fc.construct_voxel_grid(grid_res)
+        print(3.4)
         if isinstance(scale, list):
             self.verts[:, 0] = self.verts[:, 0] * scale[0]
             self.verts[:, 1] = self.verts[:, 1] * scale[1]
             self.verts[:, 2] = self.verts[:, 2] * scale[1]
         else:
             self.verts = self.verts * scale
+        print(3.5)
             
         all_edges = self.indices[:, self.fc.cube_edges].reshape(-1, 2)
         self.all_edges = torch.unique(all_edges, dim=0)
 
         # Parameters used for fix boundary sdf
+        print(3.6)
         self.center_indices, self.boundary_indices = get_center_boundary_index(self.grid_res, device)
+        print(3.7)
         self.renderer = renderer
         self.render_type = render_type
+        print(3.8)
         self.ctx = dr.RasterizeCudaContext(device=device)
+        print(3.9)
 
         # self.verts.requires_grad_(True)