dont use gfpgan

src/facerender/animate.py

@@ -20,7 +20,6 @@ from src.facerender.modules.generator import OcclusionAwareGenerator, OcclusionA
 from src.facerender.modules.make_animation import make_animation 
 
 from pydub import AudioSegment 
-from src.utils.face_enhancer import enhancer_generator_with_len, enhancer_list
 from src.utils.paste_pic import paste_pic
 from src.utils.videoio import save_video_with_watermark
 
@@ -233,22 +232,22 @@ class AnimateFromCoeff():
  full_video_path = av_path 
 
  #### paste back then enhancers
- if enhancer:
- video_name_enhancer = x['video_name'] + '_enhanced.mp4'
- enhanced_path = os.path.join(video_save_dir, 'temp_'+video_name_enhancer)
- av_path_enhancer = os.path.join(video_save_dir, video_name_enhancer) 
- return_path = av_path_enhancer
-
- try:
- enhanced_images_gen_with_len = enhancer_generator_with_len(full_video_path, method=enhancer, bg_upsampler=background_enhancer)
- imageio.mimsave(enhanced_path, enhanced_images_gen_with_len, fps=float(25))
- except:
- enhanced_images_gen_with_len = enhancer_list(full_video_path, method=enhancer, bg_upsampler=background_enhancer)
- imageio.mimsave(enhanced_path, enhanced_images_gen_with_len, fps=float(25))
+ # if enhancer:
+ # video_name_enhancer = x['video_name'] + '_enhanced.mp4'
+ # enhanced_path = os.path.join(video_save_dir, 'temp_'+video_name_enhancer)
+ # av_path_enhancer = os.path.join(video_save_dir, video_name_enhancer) 
+ # return_path = av_path_enhancer
+
+ # try:
+ # enhanced_images_gen_with_len = enhancer_generator_with_len(full_video_path, method=enhancer, bg_upsampler=background_enhancer)
+ # imageio.mimsave(enhanced_path, enhanced_images_gen_with_len, fps=float(25))
+ # except:
+ # enhanced_images_gen_with_len = enhancer_list(full_video_path, method=enhancer, bg_upsampler=background_enhancer)
+ # imageio.mimsave(enhanced_path, enhanced_images_gen_with_len, fps=float(25))
  
- save_video_with_watermark(enhanced_path, new_audio_path, av_path_enhancer, watermark= False)
- print(f'The generated video is named {video_save_dir}/{video_name_enhancer}')
- os.remove(enhanced_path)
+ # save_video_with_watermark(enhanced_path, new_audio_path, av_path_enhancer, watermark= False)
+ # print(f'The generated video is named {video_save_dir}/{video_name_enhancer}')
+ # os.remove(enhanced_path)
 
  os.remove(path)
  os.remove(new_audio_path)

src/facerender/pirender_animate.py

@@ -17,7 +17,6 @@ from src.facerender.pirender.config import Config
 from src.facerender.pirender.face_model import FaceGenerator
 
 from pydub import AudioSegment 
-from src.utils.face_enhancer import enhancer_generator_with_len, enhancer_list
 from src.utils.paste_pic import paste_pic
 from src.utils.videoio import save_video_with_watermark
 
@@ -106,22 +105,22 @@ class AnimateFromCoeff_PIRender():
  full_video_path = av_path 
 
  #### paste back then enhancers
- if enhancer:
- video_name_enhancer = x['video_name'] + '_enhanced.mp4'
- enhanced_path = os.path.join(video_save_dir, 'temp_'+video_name_enhancer)
- av_path_enhancer = os.path.join(video_save_dir, video_name_enhancer) 
- return_path = av_path_enhancer
-
- try:
- enhanced_images_gen_with_len = enhancer_generator_with_len(full_video_path, method=enhancer, bg_upsampler=background_enhancer)
- imageio.mimsave(enhanced_path, enhanced_images_gen_with_len, fps=float(25))
- except:
- enhanced_images_gen_with_len = enhancer_list(full_video_path, method=enhancer, bg_upsampler=background_enhancer)
- imageio.mimsave(enhanced_path, enhanced_images_gen_with_len, fps=float(25))
+ # if enhancer:
+ # video_name_enhancer = x['video_name'] + '_enhanced.mp4'
+ # enhanced_path = os.path.join(video_save_dir, 'temp_'+video_name_enhancer)
+ # av_path_enhancer = os.path.join(video_save_dir, video_name_enhancer) 
+ # return_path = av_path_enhancer
+
+ # try:
+ # enhanced_images_gen_with_len = enhancer_generator_with_len(full_video_path, method=enhancer, bg_upsampler=background_enhancer)
+ # imageio.mimsave(enhanced_path, enhanced_images_gen_with_len, fps=float(25))
+ # except:
+ # enhanced_images_gen_with_len = enhancer_list(full_video_path, method=enhancer, bg_upsampler=background_enhancer)
+ # imageio.mimsave(enhanced_path, enhanced_images_gen_with_len, fps=float(25))
  
- save_video_with_watermark(enhanced_path, new_audio_path, av_path_enhancer, watermark= False)
- print(f'The generated video is named {video_save_dir}/{video_name_enhancer}')
- os.remove(enhanced_path)
+ # save_video_with_watermark(enhanced_path, new_audio_path, av_path_enhancer, watermark= False)
+ # print(f'The generated video is named {video_save_dir}/{video_name_enhancer}')
+ # os.remove(enhanced_path)
 
  os.remove(path)
  os.remove(new_audio_path)