From e5e79fbde32cf96720d484511fbae2272144beb6 Mon Sep 17 00:00:00 2001 From: Xintao Date: Sun, 18 Sep 2022 23:57:35 +0800 Subject: [PATCH] deal with flv format --- inference_realesrgan_video.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/inference_realesrgan_video.py b/inference_realesrgan_video.py index 39fed08..3b38369 100644 --- 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