图像去模糊:跑通DeblurGAN-v2
目录
一.环境的配置
二.跑通测试predict.py
三.跑通训练train.py
1.数据准备
2.数据增强方式
3.加载预训练模型
工程:https://github.com/VITA-Group/DeblurGANv2
一.环境的配置
直接使用python train.py缺什么库,就安装什么库;
二.跑通测试predict.py
需要设置以下参数:
--img_pattern
/media/XXX/test/LR/2021-01-28_11-21-04_white.jpg
--mask_pattern
None
--weights_path
/media/XXX/deblur/DeblurGANv2-master/weights/fpn_inception.h5
--out_dir
submit/
--side_by_side
False
--video
False
三.跑通训练train.py
配置config中的config.yaml参数:
---
project: deblur_gan
exper
共有 0 条评论