关于跑图像去雾算法DCPDN的教程及Bug解决
内容导读
互联网集市收集整理的这篇技术教程文章主要介绍了关于跑图像去雾算法DCPDN的教程及Bug解决,小编现在分享给大家,供广大互联网技能从业者学习和参考。文章包含3204字,纯文字阅读大概需要5分钟。
内容图文
![关于跑图像去雾算法DCPDN的教程及Bug解决](/upload/InfoBanner/zyjiaocheng/590/c28ee25014a3451689d57df8335d58e1.jpg)
前情提要
??最近刚刚开始图像去雾方面的研究,自然少不了阅读这一领域的经典文献和GitHub源码。DCPDN是其中比较很有价值的一篇,在阅读文献过程中,希望跑通它的代码,结合代码来帮助我理解这一算法的实现原理。但是我在配置环境跑程序的过程中出现了许多问题,花费了许多时间和精力解决了其中部分问题,因此想在此记录下来,同时也希望对同样遇到这些问题的你有所帮助,谢谢~
论文:Densely Connected Pyramid Dehazing Network
github源码:https://github.com/hezhangsprinter/DCPDN
参考的相关博客:1. 一步一步教你跑DCPDN深度学习去雾网络
????????2. DCPDN项目
代码运行环境
- Ubuntu 18.04.3
- python 3.6
- torch 0.3.1
- torchvision 0.2.1
直接把我配环境的命令行语句贴出来吧
- 新建conda环境
conda create -n xxx(你的环境名) python=3.6
- 进入新建的环境(激活环境)
conda activate xxx(你的环境名)
- 开始安装各种依赖的包
pip install https://download.pytorch.org/whl/cu90/torch-0.3.1-cp36-cp36m-linux_x86_64.whl
pip install torchvision==0.2.1
pip install h5py
pip install scipy
(没记错的话应该就是上面这些,比较重要的是torch版本和torchvision版本,因为作者的代码是基于早期的低版本,由于新版本做了一些改动,如果不按低版本来装的话,会遇到更多麻烦的问题,亲身经历。等自己研究透了,看看能否高版本的复现一下?哈哈哈)
原始代码可能使用的是Python2版本,因此某些语法与现在的Python3不兼容,(Python2使用<>
作为“不等于”的符号,而Python3使用!=
)需要修改一下,分别位于train.py
文件的第312行、329行、353行和366行。
问题1:Missing key(s) in state_dict: xxxxxxxxxxxxx; Unexpected key(s) in state_dict: xxxxxxxxxxxxx
报错原因: 预训练模权重的字典关键字与所创建的网络模型的字典关键字不匹配。(简单来说,我们的模型要使用现有的预训练权重来进行参数的初始化,这个过程需要两者各层级的网络名称相对应,否则就会出现上述错误。)
问题出在train.py文件的第124行:
if opt.netG != '':
netG.load_state_dict(torch.load(opt.netG))
opt.netG是预训练权重文件netG_epoch_8.pth的所在路径,torch.load(opt.netG)
是加载这一权重文件。这个权重文件在以前训练的时候可能还是采用旧的字典关键字,如:‘norm.1’, ‘relu.1’, ‘conv.1’, ‘norm.2’, ‘relu.2’, ‘conv.2’,但是现在的网络模型在创建时已经不再允许使用“.”了,所以需要修改预训练权重的关键字,使其与我们的网络匹配。
通过正则修改,将上面的代码修改成以下内容:(参考torchvision.models.densenet中的做法)
if opt.netG != '':
checkpoint = torch.load(opt.netG)
pattern = re.compile(r'^(.*denselayer\d+\.(?:norm|relu|conv))\.((?:[12])\.'
r'(?:weight|bias|running_mean|running_var))$')
for key in list(checkpoint.keys()):
res = pattern.match(key)
if res:
new_key = res.group(1) + res.group(2)
checkpoint[new_key] = checkpoint[key]
del checkpoint[key]
netG.load_state_dict(checkpoint)
经过修改后,这一段代码就能顺利执行了。
问题2:"python3.6/site-packages/torch/utils/data/dataloader.py", line 271, in __next__ raise StopIteration
关于这个问题,上面参考的博客2中作出了解答。因为博客1的作者训练网络时使用以下的命令:
python train.py --dataroot ./facades/train512 --valDataroot ./facades/test512 --exp ./checkpoints_new --netG ./demo_model/netG_epoch_8.pth
其中,--valDataroot
传入的是./facades/test512
这个路径,但是源代码的作者并没有提供这一文件,只有./facades/val512
,因此把命令改成:
python train.py --dataroot ./facades/train512 --valDataroot ./facades/val512 --exp ./checkpoints_new --netG ./demo_model/netG_epoch_8.pth
就能成功解决这个问题。
最后,再次感谢上面两位博主的博客~
内容总结
以上是互联网集市为您收集整理的关于跑图像去雾算法DCPDN的教程及Bug解决全部内容,希望文章能够帮你解决关于跑图像去雾算法DCPDN的教程及Bug解决所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错,欢迎将互联网集市网站推荐给程序员好友。
内容备注
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 gblab@vip.qq.com 举报,一经查实,本站将立刻删除。
内容手机端
扫描二维码推送至手机访问。