浏览代码

add video_pose api

dongyuanjushi 4 年之前
父节点
当前提交
042cdc9a80
共有 1 个文件被更改,包括 4 次插入4 次删除
  1. 4 4
      backend/demo_func.py

+ 4 - 4
backend/demo_func.py

@@ -16,15 +16,15 @@ from video_pose.run import run_video_reconstruction
 
 
 def run_3d_for_video(original_video_path):
-    # run_3d(original_video_path)
+    run_3d(original_video_path)
     video_analyzed_path = original_video_path + ".npz"
     print(video_analyzed_path)
-    # custom(video_analyzed_path)
+    custom(video_analyzed_path)
     compressed_analyzed_path = video_analyzed_path.replace(".npz", "_compressed.npz")
     run_video_reconstruction(original_video_path, compressed_analyzed_path)
     reconstructed_video_path = original_video_path.replace(".mp4", "_reconstructed.mp4")
     print(reconstructed_video_path)
-    # return reconstructed_video_path
+    return reconstructed_video_path
 
 
 def run_openpose_for_image_front(index):
@@ -152,5 +152,5 @@ def run_openpose_for_normal(index):
 
 if __name__ == '__main__':
     # run_openpose_for_normal(1)
-    input_video = "video/13.mp4"
+    input_video = "video/15.mp4"
     run_3d_for_video(input_video)