deal with flv format

xinntao · Sep 18, 2022 · e5e79fb · e5e79fb
1 parent e5763af
commit e5e79fb
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/inference_realesrgan_video.py b/inference_realesrgan_video.py
@@ -348,6 +348,11 @@ def main():
     else:
         is_video = False
 
+    if is_video and args.input.endswith('.flv'):
+        mp4_path = args.input.replace('.flv', '.mp4')
+        os.system(f'ffmpeg -i {args.input} -codec copy {mp4_path}')
+        args.input = mp4_path
+
     if args.extract_frame_first and not is_video:
         args.extract_frame_first = False