faster rendering

body_renderer.py

@@ -1,3 +1,20 @@
+import torch
+
+def subsample_tensor(tensor, original_fps=30, target_fps=25):
+    # Total number of original frames
+    total_frames = tensor.shape[0]
+    
+    # Calculate new number of frames
+    new_frames = int(total_frames * (target_fps / original_fps))
+    
+    # Create indices for subsampling
+    indices = torch.linspace(0, total_frames - 1, new_frames).long()
+    
+    # Select frames
+    subsampled_tensor = tensor[indices]
+    
+    return subsampled_tensor
+
 def get_render(body_model_loaded, 
                body_trans, 
                body_orient, body_pose, 
@@ -21,14 +38,15 @@ def get_render(body_model_loaded,
                                         pose)
 
         vertices=vertices.vertices
+        vertices = subsample_tensor(vertices, original_fps=30, target_fps=25)
         vertices = vertices.detach().cpu().numpy()
         vertices_list.append(vertices)
 
     #Initialising the renderer
     from renderer.humor import HumorRenderer
-    fps = 30.0
-    imw = 720 # 480
-    imh = 720 # 360
+    fps = 25.0
+    imw = 480 # 480
+    imh = 360 # 360
     renderer = HumorRenderer(fps=fps, imw=imw, imh=imh)   
     
     if len(vertices_list)==2:

renderer/humor.py

@@ -14,7 +14,7 @@ FACES = torch.from_numpy(np.int32(np.load(FACE_PATH)))
 
 
 class HumorRenderer:
-    def __init__(self, fps=20.0, **kwargs):
+    def __init__(self, fps=30.0, **kwargs):
         self.kwargs = kwargs
         self.fps = fps