neverwinHao / WSR

WSRNet在1w8CelebA上的复现仓库

WSRNet复现及预训练

A pytorch implementation of Paper "Wavelet-srnet: A wavelet-based cnn for multi-scale face super resolution"

Train

CUDA_VISIBLE_DEVICES=1 python main.py --ngpu=1 --test --start_epoch=0  --test_iter=1000  --batchSize=64 --test_batchSize=32 --nrow=4  --upscale=3 --input_height=128 --output_height=128 --crop_height=128 --lr=2e-4  --nEpochs=500 --cuda

Test

CUDA_VISIBLE_DEVICES=0 python test.py --ngpu=1 --test --test_iter=1000 --test_batchSize=32 --nrow=4 --upscale=4 --input_height=128 --output_height=128 --cuda