调试Ufromer代码Uformer: A General U-Shaped Transformer for Image Restoration (CVPR 2022)代码调试

Uformer: A General U-Shaped Transformer for Image Restoration (CVPR 2022)代码调试

code:github
paper:paper

想把它运用低光图像的数据集用于低光图像增强
首先pip3 install -r requirements.txt
先下载作者得数据集调试代码看看
数据集下载
因为只是想跑通debug看看 选择了个最小的数据集500多M大小的RealBlur_R数据集,其他都挺大。
将下载得数据集放在…/datasets/deblurring/
在这里插入图片描述
这个数据集里应该只是作为测试数据集,各位道友如果想训练或者是测试,还是要下载Gopro的数据集,因为我只是想跑通debug看看,所以找了个小数据集。
关于sh命令的debug我是真的很无语,呜呜呜怎么debug啊!
于是只能先在我的低光LOL数据集上跑跑看,更改一下.sh文件里的路径还有dataloader里的路径。
更改sh路径
我的train数据集在our485里 validate数据集在LOLv2/test 里
然后更改dataloader里的路径
我这边选择了去模糊的dataloader代码 感觉直接搬到低光图像 应该会有问题
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
训练集low 是低光图像所在文件夹的名字 high是低光图像对应GT文件夹的名字 在dataloaderTrain做路径替换。
测试集Low是低光图像所在文件夹的名字 Normal是低光图像对应GT文件夹的名字,在Val_deblur做路径替换。
之后直接跑,看看效果如何。

  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值