调试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
先下载作者得数据集调试代码看看
调试Ufromer代码Uformer: A General U-Shaped Transformer for Image Restoration (CVPR 2022)代码调试_第1张图片
因为只是想跑通debug看看 选择了个最小的数据集500多M大小的RealBlur_R数据集,其他都挺大。
将下载得数据集放在…/datasets/deblurring/
调试Ufromer代码Uformer: A General U-Shaped Transformer for Image Restoration (CVPR 2022)代码调试_第2张图片
这个数据集里应该只是作为测试数据集,各位道友如果想训练或者是测试,还是要下载Gopro的数据集,因为我只是想跑通debug看看,所以找了个小数据集。
关于sh命令的debug我是真的很无语,呜呜呜怎么debug啊!
于是只能先在我的低光LOL数据集上跑跑看,更改一下.sh文件里的路径还有dataloader里的路径。
调试Ufromer代码Uformer: A General U-Shaped Transformer for Image Restoration (CVPR 2022)代码调试_第3张图片
我的train数据集在our485里 validate数据集在LOLv2/test 里
然后更改dataloader里的路径
我这边选择了去模糊的dataloader代码 感觉直接搬到低光图像 应该会有问题
调试Ufromer代码Uformer: A General U-Shaped Transformer for Image Restoration (CVPR 2022)代码调试_第4张图片
调试Ufromer代码Uformer: A General U-Shaped Transformer for Image Restoration (CVPR 2022)代码调试_第5张图片
调试Ufromer代码Uformer: A General U-Shaped Transformer for Image Restoration (CVPR 2022)代码调试_第6张图片
训练集low 是低光图像所在文件夹的名字 high是低光图像对应GT文件夹的名字 在dataloaderTrain做路径替换。
测试集Low是低光图像所在文件夹的名字 Normal是低光图像对应GT文件夹的名字,在Val_deblur做路径替换。
之后直接跑,看看效果如何。

你可能感兴趣的:(transformer,深度学习,人工智能)