Sapphire-356 commited on
Commit
ed7eb3f
·
1 Parent(s): 2ee3801

add video examples

Browse files
Files changed (3) hide show
  1. app.py +3 -3
  2. input_videos/kun_test_5sec.mp4 +0 -0
  3. videopose_PSTMO.py +3 -1
app.py CHANGED
@@ -6,14 +6,14 @@ import os
6
  def Video2MC(video, progress = gr.Progress(track_tqdm=True)):
7
 
8
  progress(1.0, desc="Step 0: Starting")
9
- output_path = gr_video2mc(video, progress)
10
 
11
- return output_path, output_path
12
 
13
 
14
  iface = gr.Interface(fn=Video2MC,
15
  inputs=gr.Video(),
16
- outputs=["file", "text"],
17
  examples=[os.path.join(os.path.dirname(__file__),
18
  "input_videos/kun_test_5sec.mp4")],
19
  )
 
6
  def Video2MC(video, progress = gr.Progress(track_tqdm=True)):
7
 
8
  progress(1.0, desc="Step 0: Starting")
9
+ output_path, output_video = gr_video2mc(video, progress)
10
 
11
+ return output_path, output_path, output_video
12
 
13
 
14
  iface = gr.Interface(fn=Video2MC,
15
  inputs=gr.Video(),
16
+ outputs=["file", "text", gr.Video()],
17
  examples=[os.path.join(os.path.dirname(__file__),
18
  "input_videos/kun_test_5sec.mp4")],
19
  )
input_videos/kun_test_5sec.mp4 ADDED
Binary file (542 kB). View file
 
videopose_PSTMO.py CHANGED
@@ -210,7 +210,9 @@ def gr_video2mc(video_path, progress):
210
  FPS_mine_imator = 30
211
  output_dir_dict = inference_video(video_path, 'alpha_pose', progress)
212
  Hk.hpe2keyframes(output_dir_dict['npy'], FPS_mine_imator, f"output_miframes/{output_dir_dict['video_name']}.miframes")
213
- return os.path.abspath(f"output_miframes/{output_dir_dict['video_name']}.miframes")
 
 
214
 
215
 
216
  if __name__ == '__main__':
 
210
  FPS_mine_imator = 30
211
  output_dir_dict = inference_video(video_path, 'alpha_pose', progress)
212
  Hk.hpe2keyframes(output_dir_dict['npy'], FPS_mine_imator, f"output_miframes/{output_dir_dict['video_name']}.miframes")
213
+ path1 = os.path.abspath(f"output_miframes/{output_dir_dict['video_name']}.miframes")
214
+ path2 = os.path.abspath(f"output_videos/{output_dir_dict['video_name']}.mp4")
215
+ return path1, path2
216
 
217
 
218
  if __name__ == '__main__':