pytorch
ENTRopYYY
机器不会学习
展开
-
CBAM非官方代码及官方代码
非官方代码 两大模块:class ChannelAttention(nn.Module): def __init__(self, in_planes, ratio=16): super(ChannelAttention, self).__init__() self.avg_pool = nn.AdaptiveAvgPool2d(1) self.max_pool = nn.AdaptiveMaxPool2d(1) self.fc1..原创 2020-11-05 00:36:29 · 7603 阅读 · 2 评论 -
简单高效的模型小挂件——CBAM
论文传送门: CBAM: Convolutional Block Attention Module代码(Pytorch): https://github.com/Jongchan/attention-module介绍卷积神经网络发展到现在已在三大视觉任务(分类, 检测, 分割)上取得不错的成绩. 从早期的LeNet到现在常见的带有残差结构的网络, 模型结构变得越来越深并有强大的表达能力. VGGNet堆砌跟多相同的模块而获得不错的结果; ResNet利用残差模块和远跳连接的拓扑结构创造出更深更优秀的网原创 2020-11-01 09:48:28 · 2666 阅读 · 1 评论 -
解决测试阶段显存增大问题
转自简书:解决 pytorch 在训练时由于设置了验证集导致 out of memory (同样可用于测试时减少显存占用)问题描述在跑pytorch的时候,在训练阶段监控到显存占用2.7G左右,但到了验证阶段发现占用了3.65G左右,对于我4G显存的显卡来说很容易爆掉。解决方法假设一开始训练和验证阶段如下:# 训练for i, (train_data, train_label) in...转载 2020-02-04 23:00:17 · 1854 阅读 · 0 评论 -
cv2加载图像
导入模块import cv2import numpy as npimport torchfrom torch.autograd import Valiable格式转换img = cv2.(img_path)img = np.resize(img, (32, 32, 3)img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) # BGR转RGBimg...原创 2019-11-26 22:12:16 · 320 阅读 · 0 评论