Project tăng cường độ phân giải ảnh . Paper"Residual Dense Network for Image Super-Resolution".
- PyTorch 1.0.0
- Numpy 1.15.4
- Pillow 5.4.1
Vào file notebook SuperResolution.ipynb
Load checkpoint tại đây https://drive.google.com/drive/folders/1hD38w5_jMTr8br2DtQcxf2wt3Op6Ea89?usp=sharing
Tạo và cd vào folder Dataset Uncomment phần wget và unzip để lấy dữ liệu đã được chia thành các folder HR và LR tương ứng
Tinh chỉnh các siêu tham số
-- num-featue : số feature ở tầng Conv đầu tiên
-- growth-rate : số feature trong các DenseBlock
-- scale : hệ số phóng
-- num-blocks : số block RDB trong 1 layer
-- num-layers : số layer
-- lr : learning rate
-- patch_size : kích thước ảnh được cắt nhỏ
-- batch_size
-- epochs Thay đổi đường dẫn đến ảnh test ( trong hàm visualize_sr ) Chạy file main.py