DeblurGAN 消除运动模糊
源码地址
https://github.com/KupynOrest/DeblurGAN
下载 weight 文件(dropbox 网盘下不了的,可以找我要。
原链接的 latest_net_G.pth 文件莫名消失了,下载见下面。
链接: https://pan.baidu.com/s/1nvQQNGH 密码: 8ab2
要 latest_net_D.pth 的朋友也不少,上传一下。
链接: https://pan.baidu.com/s/14Lq2YUYoeDYRslgUbShZNQ 密码: k4j5
下载安装 pytorch
http://pytorch.org/
选择系统和安装方式
执行后,下载很慢,迅雷下好后上传。
sudo pip3 install torch-0.2.0.post3-cp35-cp35m-manylinux1_x86_64.whl
sudo pip3 install torchvision
test.py 里 SSIM 直接 import ssim 就好
NameError: name 'use_parallel' is not defined
models/test_model.py 里 的 use_parallel 替换为 True。
test 命令
python3 test.py --dataroot ./testimage --model test --dataset_mode single --learn_residual --resize_or_crop scale_width
testimage 是存放图片的目录,–resize_or_crop scale_width 是保持图片原大小。
加 scale_width 后 处理有些图片时显示如下错误
RuntimeError: invalid argument 3: sizes do not match at /pytorch/torch/lib/THC/generated/../generic/THCTensorMathPointwise.cu:217
resize_or_crop 也可以选择 false ,但是容易爆内存。内存不足的时候推荐用 resize_and_crop
去模糊效果很不错
和 SR 一起处理的效果
待解之谜:不知道怎么能跑成翻转图像,很神奇。
终于试出来了,去掉
--learn_residual
就能出现这个效果。
效果:
原图
重新 clone 代码后正常,问题可能是某个参数错了,或文件损坏。