Source code phục hồi ảnh cũ bị mờ bằng Pytorch

Hiện nay có nhiều source code mã nguồn mở để phục hồi ảnh cũ. Bài viết này giới thiệu Real-ESRGAN – mã nguồn chạy được và hoạt động hiệu quả như giới thiệu.

Nếu bạn không rành về code mà muốn sử dụng thì xem link hướng dẫn sử dụng Real-ESRGAN online ở đây:
https://viscomsolution.com/phuc-hoi-anh-bi-mo-nhoe-bang-cong-cu-online-mien-phi/

Đây là tool dành cho Anime, Wibu,… tuy nhiên phục hồi ảnh cũ xưa rất tốt. Tùy bạn muốn sử dụng vào mục đích gì.

Source code cung cấp file binary exe, Colab Notebook để bạn sử dụng. Nếu bạn muốn chạy trên PC của mình thì làm theo các bước sau:

Clone repo

Dùng git command hoặc Source tree/Tortoise Git để clone:
git clone https://github.com/xinntao/Real-ESRGAN.git

Cài đặt required packages

PC mình đang cài đặt Python 3.7.3 x64 trên Windows 10.

Cài đặt Pytorch CUDA hoặc CPU đều được, mình dùng CUDA 11.3
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

Sau đó cài đặt các packages:
pip install basicsr
pip install facexlib
pip install gfpgan
pip install -r requirements.txt
python setup.py develop

Sau khi cài đặt xong các bạn download file pretrain RealESRGAN_x4plus.pth bỏ vào folder Real-ESRGAN\experiments\pretrained_models

Run source code

python inference_realesrgan.py -n RealESRGAN_x4plus -i 4229808217108.jpg -o outfile --fp32 --face_enhance -s 1

Giải thích command:

  • -n RealESRGAN_x4plus: chọn model đã download ở trên
  • -i 4229808217108.jpg: ảnh input
  • -o outfile: folder chứa ảnh output
  • –fp32: chọn độ chính xác số float, có thể chọn –fp64 (mặc định) hoặc –fp16 nếu máy bạn không đủ VRAM
  • –face_enhance: thêm tham số này nếu ảnh input có khuôn mặt
  • -s 1: tỷ lệ scale, 1 là giữ nguyên, mặc định là 4

Kết quả sẽ lưu trong folder outfile bạn đã chọn. Một số kết quả để bạn tham khảo

Source code

https://github.com/NightmareAI/Real-ESRGAN