Update src/app/render.py

src/app/render.py

@@ -19,7 +19,7 @@ def get_smplx_model(bs, deivce):
     return smpl_model.eval()
 
 
-def render_motion_to_video(motion, device, output_path, title, step=2):
+def render_motion_to_video(motion, device, output_path, title, step=3):
     length = motion['transl'].shape[0]
     model = get_smplx_model(length, device)
     vertices = model(
@@ -61,20 +61,11 @@ def render_motion_to_video(motion, device, output_path, title, step=2):
         
         img_path = os.path.join(temp_dir, f"frame_{i:04d}.png")
         cv2.imwrite(img_path, image)
-        
-        # font = cv2.FONT_HERSHEY_SIMPLEX
-        # font_scale = 1.0
-        # font_color = (0, 55, 0)  # 白色
-        # font_thickness = 2
-        # text_size = cv2.getTextSize(title, font, font_scale, font_thickness)[0]
-        # text_x = (image.shape[1] - text_size[0]) // 2  # 水平居中
-        # text_y = image.shape[0] - 30  # 底部上方30像素
-        # cv2.putText(image, title, (text_x, text_y), font, font_scale, font_color, font_thickness)
         images.append(image)
     
     if images:
         height, width, _ = images[0].shape
-        sampled_fps = 20 / step
+        sampled_fps = int((20+step/2) / step)
         temp_video_path = output_path.replace('.mp4', '_temp.mp4')
 
         fourcc = cv2.VideoWriter_fourcc(*'mp4v')
@@ -112,8 +103,12 @@ def render_motion_to_video(motion, device, output_path, title, step=2):
 
 
 from argparse import ArgumentParser
+import time
 if __name__ == "__main__":
     parser = ArgumentParser()
+    print('******Start rendering...******')
+    start_time = time.time()
+    
     parser.add_argument("--motion_path", type=str, required=True)
     parser.add_argument("--title", type=str, default="rendered_motion.mp4")
 
@@ -125,3 +120,5 @@ if __name__ == "__main__":
 
     print(args.title)
     render_motion_to_video(motion, device, output_path, args.title)
+
+    print(f"******Rendering finished in {time.time() - start_time:.2f} seconds.******")