update for video enhance.

README.md

@@ -100,7 +100,7 @@ You can put the testing images in the `inputs/TestWhole` folder. If you would li
 🧑🏻 Face Restoration (cropped and aligned face)
 ```
 # For cropped and aligned faces
-python inference_codeformer.py -w 0.5 --has_aligned --input_path [input folder]
+python inference_codeformer.py -w 0.5 --has_aligned --input_path [image folder]|[image path]
 ```
 
 :framed_picture: Whole Image Enhancement
@@ -108,12 +108,17 @@ python inference_codeformer.py -w 0.5 --has_aligned --input_path [input folder]
 # For whole image
 # Add '--bg_upsampler realesrgan' to enhance the background regions with Real-ESRGAN
 # Add '--face_upsample' to further upsample restorated face with Real-ESRGAN
-python inference_codeformer.py -w 0.7 --input_path [image folder/image path]
+python inference_codeformer.py -w 0.7 --input_path [image folder]|[image path]
 ```
 
 :clapper: Video Enhancement
 ```
+# For Windows/Mac users, please install ffmpeg first
+conda install -c conda-forge ffmpeg
+```
+```
 # For video clips
+# video path should end with '.mp4'|'.mov'|'.avi'
 python inference_codeformer.py --bg_upsampler realesrgan --face_upsample -w 1.0 --input_path [video path]
 ```
 

basicsr/utils/video_util.py

@@ -30,10 +30,16 @@ class VideoReader:
     def __init__(self, video_path):
         self.paths = []  # for image&folder type
         self.audio = None
-        self.stream_reader = (
-            ffmpeg.input(video_path).output('pipe:', format='rawvideo', pix_fmt='bgr24',
-                                            loglevel='error').run_async(
-                                                pipe_stdin=True, pipe_stdout=True, cmd='ffmpeg'))
+        try:
+            self.stream_reader = (
+                ffmpeg.input(video_path).output('pipe:', format='rawvideo', pix_fmt='bgr24',
+                                                loglevel='error').run_async(
+                                                    pipe_stdin=True, pipe_stdout=True, cmd='ffmpeg'))
+        except FileNotFoundError:
+            print('Please install ffmpeg (not ffmpeg-python) by running\n',
+                  '\t$ conda install -c conda-forge ffmpeg')
+            sys.exit(0)
+
         meta = get_video_meta_info(video_path)
         self.width = meta['width']
         self.height = meta['height']

inference_codeformer.py

@@ -107,6 +107,9 @@ if __name__ == '__main__':
         result_root = args.output_path
 
     test_img_num = len(input_img_list)
+    if test_img_num == 0:
+        raise FileNotFoundError("\nInput file is not found.")
+
     # ------------------ set up background upsampler ------------------
     if args.bg_upsampler == 'realesrgan':
         bg_upsampler = set_realesrgan()

requirements.txt

@@ -14,5 +14,4 @@ torchvision
 tqdm
 yapf
 lpips
-gdown # supports downloading the large file from Google Drive
-ffmpeg-python
+gdown # supports downloading the large file from Google Drive