Concurrent Spatial and Channel Squeeze & Excitation in Fullly Convolutional Networks 代码复现

Miracle
1002
文章
46
评论
2019年4月18日17:22:48 评论 1297字阅读4分19秒

三种SENet Pytorch代码如下:

  1. class CSEModule(nn.Module):
  2.     def __init__(self, ch, re=16):
  3.         super().__init__()
  4.         self.cSE = nn.Sequential(nn.AdaptiveAvgPool2d(1),
  5.                                  nn.Conv2d(ch,ch//re,1),
  6.                                  nn.ReLU(inplace=True),
  7.                                  nn.Conv2d(ch//re,ch,1),
  8.                                  nn.Sigmoid())
  9.     def forward(self, x):
  10.         return x * self.cSE(x) #cSE
  11. class SSEModule(nn.Module):
  12.     def __init__(self, ch, re=16):
  13.         super().__init__()
  14.         self.sSE = nn.Sequential(nn.Conv2d(ch,ch,1),
  15.                                  nn.Sigmoid())
  16.     def forward(self, x):
  17.         return x * self.sSE(x)  #sSE
  18. class SCSEModule(nn.Module):
  19.     def __init__(self, ch, re=16):
  20.         super().__init__()
  21.         self.cSE = nn.Sequential(nn.AdaptiveAvgPool2d(1),
  22.                                  nn.Conv2d(ch,ch//re,1),
  23.                                  nn.ReLU(inplace=True),
  24.                                  nn.Conv2d(ch//re,ch,1),
  25.                                  nn.Sigmoid())
  26.         self.sSE = nn.Sequential(nn.Conv2d(ch,ch,1),
  27.                                  nn.Sigmoid())
  28.     def forward(self, x):
  29.         return x * self.cSE(x) + x * self.sSE(x)  #scSE

论文原文:Concurrent Spatial and Channel Squeeze & Excitation in Fullly Convolutional Networks

继续阅读
  • 版权声明: 发表于 2019年4月18日17:22:48
  • 转载注明:https://x1995.cn/4534.html
利用深度学习自动补全 Python 代码 干货教程

利用深度学习自动补全 Python 代码

代码补全功能在IDE里面十分常见,优秀的代码自动补全功能可以大大提升工作效率。不过, IDE 基本都使用搜索方法进行补全,在一些场景下效果不佳。今日,猿妹在GitHub上找到一个开源项目,使用深度学习...
分享8点超级有用的Python编程建议 干货教程

分享8点超级有用的Python编程建议

我们在用Python进行机器学习建模项目的时候,每个人都会有自己的一套项目文件管理的习惯,我自己也有一套方法,是自己曾经踩过的坑总结出来的,现在在这里分享一下给大家,希望多少有些地方可以给大家借鉴。 ...
匿名

发表评论

匿名网友 填写信息

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen: