deal with flv format

2022-09-18 23:57:35 +08:00
parent e5763af574
commit e5e79fbde3
1 changed files with 5 additions and 0 deletions
--- 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