Pytorch_复现DHSNet过程、问题记录

最新推荐文章于 2022-10-17 22:04:51 发布
最新推荐文章于 2022-10-17 22:04:51 发布
阅读量1k 收藏 4
点赞数 2
分类专栏： Pytorch学习笔记 Salent Object Detection 文章标签： Salient Object Detection Pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/lgzh123/article/details/88826208
版权
本文记录了作者在2019年复现DHSNet（Deep Hierarchical Saliency Network）的过程，包括数据集准备、PyTorch构建网络、计算MAE和F-Measure等。作者遇到并解决了vgg16_bn输出尺寸、全连接层设计等问题，并实现了RCL（Recurrent Convolutional Layers）。通过分阶段训练，最终在测试集上达到了约0.8的F-measure。
摘要由CSDN通过智能技术生成

Salient object detection 显著性目标检测
2019.3.11～3.22 期间复现DHSNet
由于是第一次复现显著性目标检测算法，有很多东西需要边学边写，且中间尝试了Encoder_Decoder模型并且进行了训练，所以用了将近两周，时间比较长，过程、工作、问题记录如下。

一、论文

论文标题：DHSNet: Deep Hierarchical Saliency Network for Salient Object Detection
论文链接：http://xueshu.baidu.com/usercenter/paper/show?paperid=82824289b92ac32d34f059e3125212b4&site=xueshu_se
发表在2016年的CVPR上

二、Dataset

1.下载显著性目标检测常用的几个数据集：

DUT-OMRON
MSRA10K
ECSSD
PASCAL-S
HKU-IS

2.构建用于训练DHSNet的数据集

论文中使用的训练集组成如下：

We randomly selected 6,000 images from MSRA10K dataset and 3,500 images from DUT-OMRON dataset as the training set, and another 800 images from MSRA10K and 468 images from DUT-OMRON as the validation set.

写python脚本实现数据集的构建，分别对DUT-OMRON和MSRA10K做如下操作：

1.生成范围在[0, 单个数据集样本数]之间的不重复随机数作为图片编号，用random.sample()实现
2
最低0.47元/天 解锁文章
etbrrre
关注
  • 2
    点赞
  • 踩
  • 4
    收藏
    觉得还不错? 一键收藏
  • 5
    评论
专栏目录
model.vgg16.feature[:-1]的意思
养猪的纺织工
12-10 3404
用于记录： pretrained_model = models.vgg16(True).features[:-1] print(pretrained_model) 就是把vgg16最后一层不要了 pretrained_model = models.vgg16(True).features print(pretrained_model) torch保存的模型和参数按照字典方式保存。 ...
DHSNet： Deep Hierarchical Saliency Network for Salient Object Detection阅读总结
qq_40092110的博客
04-06 1075
DHSNet重要总结 1. 主要贡献 作者提出一个端到端的深层次显著性网络DHSNet，DHSNet把整个图片当作是输入，直接输出显著性图，DHSNet分层次地从全局视角到局部内容检测目标，从粗略的尺寸到细微的尺寸。 2. 模型介绍 1.简述 DHSNet由GV-CNN(在DHSNet中，前半部分实质是去掉了VGG16的最后一个池化层得到尺寸为[512, 28, 28]的Feature Map)和...
dice，准确率（Accuracy）、精确率(Precision)、召回率(Recall)、F值(F-Measure)，PCC计算pytorch代码
weixin_45763058的博客
09-09 2562
具体的原理就不讲解了，直接上代码 label和predict都是one-hot 型 def compute_PCC(predicts,label): assert(predicts.shape==label.shape) PCC=[] for i in range(predicts.shape[1]): n = predicts.shape[0]*predicts.shape[2]*predicts.shape[3] TP = ((predicts[:
聚类f_measure 评价指标(python 实现)
只想买把蝴蝶knife
01-28 2449
def f_measure(cluster, labels): def get_set(data): products=[] for i in itertools.product(data, data): temp = list(i) temp.sort() products.append(tuple(temp)) products = [(i,j) for i,j in prod
pytorch 利用vgg16_bn进行特征提取，然后进行分类
GeneralJing的专栏
01-14 1451
构建网络： import torch import torchvision.models as models class VGG(torch.nn.Module): def __init__(self, vgg='vgg16_bn', data_set='CIFAR10', pretrained=False): super(VGG, self).__init__() self.features = models.__dict__[vgg](pretrained=
mnist.zip_MNIST pytorch_pytorch_pytorch mnist_pytorch network_zi
09-20
在"pytorch_mnist"和"pytorch_network"标签中，再次强调了我们是在PyTorch中使用网络模型处理MNIST数据。这可能意味着文件包含了一个示例，展示如何从头开始创建一个简单的神经网络模型，或者利用预定义的模型类来...
FCN的pytorch实现_pytorch_fcnpytorch_FCN模型pytorch_FCN复现_fcn_
10-01
它的提出解决了传统卷积神经网络在处理像素级预测任务时的效率问题，如语义分割和实例分割。PyTorch是一个流行的深度学习框架，它提供了灵活的接口来构建和实现各种神经网络模型，包括FCN。 在这个项目中，我们使用...
pytorch_wavelets.zip
11-26
This is a PyTorch implementation for the wavelet analysis outlined in Torrence and Compo (BAMS, 1998). The code builds upon the excellent implementation of Aaron O'Leary by adding a PyTorch filter ...
pytorch_msssim
08-15
pytorch_msssim源码
显著性检测代码及图片_显著性检测HC/LC/AC/FT算法_
10-01
实现了显著性检测HC/LC/AC/FT的C++算法，并附带测试图片
在pytorch 中计算精度、回归率、F1 score等指标的实例
09-18
今天小编就为大家分享一篇在pytorch 中计算精度、回归率、F1 score等指标的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
PyPI 官网下载 | pytorch_pretrained_bert-0.3.0.tar.gz
01-29
本文将围绕PyTorch中的预训练BERT库`pytorch_pretrained_bert`进行深入讲解，以帮助读者理解其核心概念、安装与使用方法，并探讨其在实际项目中的应用。 一、`pytorch_pretrained_bert`库介绍 `pytorch_pretrained...
显著目标检测metric总结
m0_53292725的博客
10-17 1140
【代码】显著目标检测metric总结。
DHS-net 安装记录
haocrab的记录本
08-31 243
DHS-net 命令： make clean && make all -j8 make matcaffe 遇到问题： 1 找不到hdf5 修改makefile.config # INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial/ # LIBRARY_DIRS :...
论文阅读：CVPR2016 Paper list
热门推荐
一亩半分地
05-12 8万+
CVPR2016 accepted paper list
各种对象检测论文总结(Object Detection )
junmuzi的专栏
12-01 2万+
Original url: http://blog.csdn.net/u010167269/article/details/52563573 https://handong1587.github.io/deep_learning/2015/10/09/object-detection.html Object Detection  Published: 09 Oc
基于SV简单的数字IC验证框架搭建
wczzmmp2的博客
07-17 6034
本文基于systemverilog搭建一个简单的验证框架(框架图如下所示)，对于ic验证小白的入门指导。
DUT——同相运算放大器
SsyPuKoo的博客
05-06 1984
基于小豪豪老师讲述的同相运算放大器的uvm项目，加以自己的理解简单阐述，主要目的为了做一个学习记录。参考: https://blog.csdn.net/howard789/article/details/116479260
pytorch安装pytorch_grad_cam
最新发布
05-17
要安装pytorch_grad_cam，可以通过以下步骤进行操作： 1. 确认已安装PyTorch和OpenCV。可以通过以下命令来检查： ``` import torch import cv2 print(torch.__version__) print(cv2.__version__) ``` 2. 安装...

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

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

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值