当前位置:主页 > 查看内容

[代码]利用python+pytorch复现Cascaded Partial Decoder for Fas

发布时间:2021-05-29 00:00| 位朋友查看

简介:本文适用于初学者进行代码复现便于初学者更好地体会代码以及论文的含义。 原论文Cascaded Partial Decoder for Fast and Accurate Salient Object DetectionCPD 原文链接代码 代码 下载好代码后下载数据集ECSSD和DUTS-TE选择下载两个将两个数据集放入文件夹……

本文适用于初学者进行代码复现,便于初学者更好地体会代码以及论文的含义。

原论文:Cascaded Partial Decoder for Fast and Accurate Salient Object Detection(CPD)

原文链接代码:代码
下载好代码后下载数据集ECSSD和DUTS-TE(选择下载两个)将两个数据集放入文件夹中。(文件夹后文会展示)
再在原文链接中下载
在这里插入图片描述
放入\CPD-master文件夹下。
将vgg.py文件内的

 # pre_train = torch.load('G:/Github/Saliency/CPD-master/model/vgg16-397923af.pth')
        # self._initialize_weights(pre_train)

两行注释掉。(这两行用于预训练)
然后修改代码路径:

dataset_path = 'E:/pytorchlearn/CPD-master/CPD-master/data/ECSSD'
dataset_path = './data/'

这两种写成哪一种都可以(对于初学者来说路径真是令人头疼)。
然后数据集这里改成:

test_datasets = ['DUTS-TE','ECSSD']

然后将数据集放在上述路径下,并更更改mask和images的名字
在这里插入图片描述
在这里插入图片描述
由于scipy.misc无法调用imsave,经查找,使用opencv进行图片保存。
在anaconda ptompt中,输入conda activate pytorch激活pytorch,然后输入下列命令即可安装opencv(由于我pytorch就是使用conda安装因此opencv也使用conda安装)

conda install -c https://conda.binstar.org/menpo opencv

然后将imsave代码行注释掉使用cv2.imwrite存储图片,在代码最开始也要加import cv2

 # misc.imsave(save_path+name, res)
 cv2.imwrite(save_path+name, res)

然后运行test_CPD.py文件,运行之前要在pycharm中配置好环境。
原mask(DUTS-TE数据集)
原mask
在这里插入图片描述
上图为预测结果。
可见在某些图片预测结果和原mask还是存在一些差距。
以上是利用已经训练好的模型进行预测,如果想自己训练可运行train.py文件进行训练。
接下来的几篇我可能会写论文的学习笔记。还是有关显著性检测方面的。

;原文链接:https://blog.csdn.net/weixin_45016866/article/details/115582161
本站部分内容转载于网络,版权归原作者所有,转载之目的在于传播更多优秀技术内容,如有侵权请联系QQ/微信:153890879删除,谢谢!
上一篇:python爬虫--爬取网易云音乐评论 下一篇:没有了

推荐图文


随机推荐