pytorch SENet实现案例


Posted in Python onJune 24, 2020

我就废话不多说了,大家还是直接看代码吧~

from torch import nn

class SELayer(nn.Module):
 def __init__(self, channel, reduction=16):
  super(SELayer, self).__init__()

  //返回1X1大小的特征图,通道数不变
  self.avg_pool = nn.AdaptiveAvgPool2d(1)
  self.fc = nn.Sequential(
   nn.Linear(channel, channel // reduction, bias=False),
   nn.ReLU(inplace=True),
   nn.Linear(channel // reduction, channel, bias=False),
   nn.Sigmoid()
  )

 def forward(self, x):
  b, c, _, _ = x.size()

  //全局平均池化,batch和channel和原来一样保持不变
  y = self.avg_pool(x).view(b, c)

  //全连接层+池化
  y = self.fc(y).view(b, c, 1, 1)

  //和原特征图相乘
  return x * y.expand_as(x)

补充知识:pytorch 实现 SE Block

论文模块图

pytorch SENet实现案例

代码

import torch.nn as nn
class SE_Block(nn.Module):
 def __init__(self, ch_in, reduction=16):
  super(SE_Block, self).__init__()
  self.avg_pool = nn.AdaptiveAvgPool2d(1)				# 全局自适应池化
  self.fc = nn.Sequential(
   nn.Linear(ch_in, ch_in // reduction, bias=False),
   nn.ReLU(inplace=True),
   nn.Linear(ch_in // reduction, ch_in, bias=False),
   nn.Sigmoid()
  )

 def forward(self, x):
  b, c, _, _ = x.size()
  y = self.avg_pool(x).view(b, c)
  y = self.fc(y).view(b, c, 1, 1)
  return x * y.expand_as(x)

现在还有许多关于SE的变形,但大都大同小异

以上这篇pytorch SENet实现案例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python中遍历文件的3个方法
Sep 02 Python
详解python之简单主机批量管理工具
Jan 27 Python
python的构建工具setup.py的方法使用示例
Oct 23 Python
基于Python中单例模式的几种实现方式及优化详解
Jan 09 Python
浅谈Python2、Python3相对路径、绝对路径导入方法
Jun 22 Python
Python字典中的键映射多个值的方法(列表或者集合)
Oct 17 Python
python实现中文文本分句的例子
Jul 15 Python
django组合搜索实现过程详解(附代码)
Aug 06 Python
Python之字典对象的几种创建方法
Sep 30 Python
python利用platform模块获取系统信息
Oct 09 Python
python利用proxybroker构建爬虫免费IP代理池的实现
Feb 21 Python
Python数据分析入门之数据读取与存储
May 13 Python
利用PyTorch实现VGG16教程
Jun 24 #Python
python安装读取grib库总结(推荐)
Jun 24 #Python
Pytorch mask-rcnn 实现细节分享
Jun 24 #Python
在Pytorch中使用Mask R-CNN进行实例分割操作
Jun 24 #Python
OpenCV+python实现实时目标检测功能
Jun 24 #Python
基于Python下载网络图片方法汇总代码实例
Jun 24 #Python
Python 分布式缓存之Reids数据类型操作详解
Jun 24 #Python
You might like
PHP最常用的ini函数分析 针对PHP.ini配置文件
2010/04/22 PHP
MySQL的FIND_IN_SET函数使用方法分享
2012/03/27 PHP
php计算整个mysql数据库大小的方法
2015/06/19 PHP
CI框架中$this->load->library()用法分析
2016/05/18 PHP
一个简单安全的PHP验证码类 附调用方法
2016/06/24 PHP
PHP实现的redis主从数据库状态检测功能示例
2017/07/20 PHP
php 使用mpdf实现指定字段配置字体样式的方法
2019/07/29 PHP
有关于eclipse配置spket需要注意的一些地方
2013/04/07 Javascript
在Iframe中获取父窗口中表单的值(示例代码)
2013/11/22 Javascript
Js实现动态添加删除Table行示例
2014/04/14 Javascript
jquery拖拽排序简单实现方法(效果增强版)
2016/02/16 Javascript
浅析JavaScript中命名空间namespace模式
2016/06/22 Javascript
JS使用插件cryptojs进行加密解密数据实例
2017/05/11 Javascript
jQuery图片缩放插件smartZoom使用实例详解
2017/08/25 jQuery
jQuery中元素选择器(element)简单用法示例
2018/05/14 jQuery
vue环形进度条组件实例应用
2018/10/10 Javascript
js获取本日、本周、本月的时间代码
2020/02/01 Javascript
Python实现字符串格式化的方法小结
2017/02/20 Python
linux下python使用sendmail发送邮件
2018/05/22 Python
python爬虫实例详解
2018/06/19 Python
对python生成业务报表的实例详解
2019/02/03 Python
基于Python的PIL库学习详解
2019/05/10 Python
python Matplotlib底图中鼠标滑过显示隐藏内容的实例代码
2019/07/31 Python
Django单元测试工具test client使用详解
2019/08/02 Python
Python爬虫防封ip的一些技巧
2020/08/06 Python
Python调用系统命令os.system()和os.popen()的实现
2020/12/31 Python
美国鞋类购物网站:Shiekh Shoes
2016/08/21 全球购物
一份全面的PHP面试问题考卷
2012/07/15 面试题
学生请假条
2014/04/11 职场文书
学校领导班子成员查摆问题及整改措施
2014/10/28 职场文书
会议室使用管理制度
2015/08/06 职场文书
医院病假条范文
2015/08/17 职场文书
小学一年级语文教学反思
2016/03/03 职场文书
OpenCV3.3+Python3.6实现图片高斯模糊
2021/05/18 Python
使用Djongo模块在Django中使用MongoDB数据库
2021/06/20 Python
win10滚动条自动往上跑怎么办?win10滚动条自动往上跑的解决方法
2022/08/05 数码科技