speed up rendering

data/pose_dataset.py

@@ -76,6 +76,9 @@ class Dataset(Dataset):
         body = self.smpl(pose.unsqueeze(0), shape.unsqueeze(0), remove=False)
         body_out = self.smpl(pose_out.unsqueeze(0), shape.unsqueeze(0), remove=False)
 
+        # print(f'smpl took {time() - start} seconds')
+        # start = time()
+
         render = self._render_body(body[0])[0,:,:,:3]
         render_out = self._render_body(body_out[0])[0,:,:,:3]
 
@@ -92,7 +95,7 @@ class Dataset(Dataset):
         render = render * mask
         render_out = render_out * mask_out
 
-        # print(f'one frame took {time() - start} seconds')
+        # print(f'render took {time() - start} seconds')
 
         data = { 
                  'pose': render,

data/pose_renderer.py

@@ -47,12 +47,14 @@ class PoseRenderer(Module):
                             image_size=(512, 512), 
                             blur_radius=0.0, 
                             faces_per_pixel=1, 
-                            bin_size=0
+                            bin_size=None,
+                            max_faces_per_bin=None,
                         )
         
         renderer = MeshRenderer(
             rasterizer=MeshRasterizer(cameras=cameras, raster_settings=raster_settings),
             shader=HardPhongShader(device=verts.device, cameras=cameras, lights=AmbientLights(device=verts.device))
+            # shader=SoftPhongShader(device=verts.device, cameras=cameras, lights=AmbientLights(device=verts.device))
         )
 
         output = renderer(Meshes(verts[None,:,:], faces[None,:,:], textures=self.textures))