dongyuanjushi 4 лет назад
Родитель
Сommit
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)