深度学习
文章平均质量分 65
dididada~
在读研究生
展开
-
数据集读取中的Image.open与cv2.imread区别在哪里?
最近在跑PSPNet语义分割的网络时,用作者给的源代码运行出现错误。错误为:/opt/conda/conda-bld/pytorch_1565272271120/work/aten/src/THCUNN/SpatialClassNLLCriterion.cu:103: void cunn_SpatialClassNLLCriterion_updateOutput_kernel(T *, T *, T *, long *, T *, int, int, int, int, int, long) [with原创 2021-01-12 16:36:33 · 650 阅读 · 0 评论 -
Paddle图像分割7日打卡营学习总结
Paddle图像分割7日打卡营学习总结1.FCN1.1 什么是FCN?1.2 FCN网络结构1.3 feature map上采样2. U-Net3. PSPNet4. DeepLab5.总结1.FCNFCN论文地址:https://arxiv.org/abs/1411.4038代码地址:https://github.com/shelhamer/fcn.berkeleyvision.org1.1 什么是FCN?对于一般的分类CNN网络,都会在网络的最后加入全连接层,经过softmax后就可以获得类别原创 2020-10-26 10:27:08 · 341 阅读 · 1 评论 -
pointnet代码复现(一)
目录1.前提准备2.pointnet分类网络1.前提准备首先说一下我的电脑配置软件:linux16.04anaconda+python3.6+tensorflow1.14硬件:cpu 8GB根据官方文档:pointnet.github需要先安装h5py:sudo apt-get install libhdf5-devsudo pip install h5py然后下载pointnet源码,下面具体讲复现步骤。2.pointnet分类网络2.1 训练分类用的数据集为modeln原创 2020-09-19 19:33:55 · 2664 阅读 · 7 评论 -
Pointnet语义分割任务在S3DIS数据集上复现
目录标题1.环境搭建2.训练三级目录1.环境搭建ubuntu16.04 双显卡nvidia gtx1080/quadro M4000anaconda3创建虚拟环境 python=3.7cuda10.0 cudnn7.65tensorflow-gpu1.14 安装命令 ——>pip install -i https://pypi.mirrors.ustc.edu.cn/simple/ tensorflow-gpu==1.14.02.训练这里主要是pointnet的语义分割代码原创 2020-09-14 10:50:45 · 3641 阅读 · 20 评论 -
注意力机制CBAM代码实现(续篇)
点这里查看上篇文章Github:kobiso/CBAM-tensorflow下面这个代码是再通道注意力模块时,将平均和最大池化后得到得描述子分别输入MLP,从代码中也可以看到连个描述子得MLP权重共享。def cbam_block(input_feature, index, reduction_ratio=8): with tf.variable_scope('cbam_%s' % index): attention_feature = channel_attention(in原创 2020-05-11 22:17:01 · 7496 阅读 · 9 评论 -
注意力机制CBAM实现
文章目录1.CBAM简介1.CBAM简介论文:CBAM: Convolutional Block AttentionModule提出了一个简单但有效的注意力模块 CBAM,给定一个中间特征图,沿着空间和通道两个维度依次推断出注意力权重,然后与原特征图相乘来对特征进行自适应调整。由于 CBAM 是一个轻量级的通用模块,它可以无缝地集成到任何 CNN 架构中,额外开销忽略不计,并且可以与基本 CNN 一起进行端到端的训练。在不同的分类和检测数据集上,将 CBAM 集成到不同的模型中后,模型的表现都有了一原创 2020-05-11 17:19:50 · 7287 阅读 · 3 评论