Требование:
- Ubuntu (22.04 протестировано) с установленными графическими процессорами Nvidia и CUDA
- Python >= 3.7 (Рекомендуется использовать Anaconda или Miniconda)
- PyTorch >= 1.7
Установка:
- Клонировать репозиторий:
git clone https://github.com/TencentARC/GFPGAN.git
cd GFPGAN
2. Установить зависимые пакеты:
pip install basicsr
pip install facexlib
pip install -r requirements.txt
python setup.py develop
pip install realesrgan
3. Скачать предварительно обученные модели:
wget https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth -P experiments/pretrained_models
Как пользоваться:
Помещаем в папку inputs/whole_imgs старые фотографии (у меня до 10 штук обрабатывало, зависит от железа и видеокарты).
Запускаем сначала conda, потом следующую команду:
python inference_gfpgan.py -i inputs/whole_imgs -o results -v 1.3 -s 2
обработанные фото ищем в папке:
results/restored_imgs
В рамках проекта: моёнаследие.рф помогу с реставрацией старых фотографий.