add gdown in requirements.

README.md

@@ -22,11 +22,15 @@ S-Lab, Nanyang Technological University
 
 - **2022.08.23**: Some modifications on face detection and fusion for better AI-created face enhancement.
 - **2022.08.07**: Integrate Real-ESRGAN to support background image enhancement.
-- **2022.07.29**: New face detector with supporting `['YOLOv5', 'RetinaFace']`. 
-- **2022.07.17**: The Colab demo of CodeFormer is available now. <a href="https://colab.research.google.com/drive/1m52PNveE4PBhYrecj34cnpEeiHcC5LTb?usp=sharing"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="google colab logo"></a>
+- **2022.07.29**: Integrate new face detectors of `['RetinaFace'(default), 'YOLOv5']`. 
+- **2022.07.17**: Colab demo of CodeFormer is available now. <a href="https://colab.research.google.com/drive/1m52PNveE4PBhYrecj34cnpEeiHcC5LTb?usp=sharing"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="google colab logo"></a>
 - **2022.07.16**: Test code for face restoration is released. :blush:
 - **2022.06.21**: This repo is created.
 
+## TODO
+- [ ] Add checkpoint for face inpainting
+- [ ] Add training code and config files
+- [x] ~~Add background image enhancement~~
 
 #### Face Restoration
 

facelib/utils/misc.py

@@ -4,6 +4,7 @@ import os.path as osp
 import torch
 from torch.hub import download_url_to_file, get_dir
 from urllib.parse import urlparse
+# from basicsr.utils.download_util import download_file_from_google_drive
 import gdown
 
 
@@ -21,6 +22,7 @@ def download_pretrained_models(file_ids, save_path_root):
             if user_response.lower() == 'y':
                 print(f'Covering {file_name} to {save_path}')
                 gdown.download(file_url, save_path, quiet=False)
+                # download_file_from_google_drive(file_id, save_path)
             elif user_response.lower() == 'n':
                 print(f'Skipping {file_name}')
             else:
@@ -28,6 +30,7 @@ def download_pretrained_models(file_ids, save_path_root):
         else:
             print(f'Downloading {file_name} to {save_path}')
             gdown.download(file_url, save_path, quiet=False)
+            # download_file_from_google_drive(file_id, save_path)
 
 
 def imwrite(img, file_path, params=None, auto_mkdir=True):

requirements.txt

@@ -14,7 +14,7 @@ torchvision
 tqdm
 yapf
 lpips
-# gdown # supports downloading the large file from Google Drive
+gdown # supports downloading the large file from Google Drive
 # cmake
 # dlib
 # conda install -c conda-forge dlib