刘虹雨 commited on
Commit
216e4f3
·
1 Parent(s): 0f2ea6a

update code

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -621,9 +621,9 @@ def avatar_generation(items, save_path_base, video_path_input, source_type, is_s
621
  # Save output images
622
  frame_name = f'{str(frame_index).zfill(4)}.png'
623
  Image.fromarray(depth, 'RGB').save(os.path.join(save_frames_path_depth, frame_name))
624
- # Image.fromarray(final_out_notshow, 'RGB').save(os.path.join(save_frames_path_out, frame_name))
625
 
626
- Image.fromarray(final_out_show, 'RGB').save(os.path.join(save_frames_path_outshow, frame_name))
627
 
628
  # Generate videos
629
  images_to_video(save_frames_path_out, os.path.join(save_path_base, image_name + video_name+ '_out.mp4'))
 
621
  # Save output images
622
  frame_name = f'{str(frame_index).zfill(4)}.png'
623
  Image.fromarray(depth, 'RGB').save(os.path.join(save_frames_path_depth, frame_name))
624
+ Image.fromarray(final_out_show, 'RGB').save(os.path.join(save_frames_path_out, frame_name))
625
 
626
+ # Image.fromarray(final_out_show, 'RGB').save(os.path.join(save_frames_path_outshow, frame_name))
627
 
628
  # Generate videos
629
  images_to_video(save_frames_path_out, os.path.join(save_path_base, image_name + video_name+ '_out.mp4'))