speed up rendering

2023-06-08 11:27:01 +00:00
parent 882ddf5578
commit 1b363a820e
2 changed files with 7 additions and 2 deletions
@@ -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,