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

Miracle
Miracle
Miracle
1357
文章
109
评论
2019年4月18日17:22:48干货教程 深度学习Concurrent Spatial and Channel Squeeze & Excitation in Fullly Convolutional Networks 代码复现已关闭评论2,38481297字阅读4分19秒阅读模式

三种SENet Pytorch代码如下：文章源自联网快讯-https://x1995.cn/4534.html

  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
  13. class SSEModule(nn.Module):
  14.     def __init__(self, ch, re=16):
  15.         super().__init__()
  16.         self.sSE = nn.Sequential(nn.Conv2d(ch,ch,1),
  17.                                  nn.Sigmoid())
  18.     def forward(self, x):
  19.         return x * self.sSE(x)  #sSE
  22. class SCSEModule(nn.Module):
  23.     def __init__(self, ch, re=16):
  24.         super().__init__()
  25.         self.cSE = nn.Sequential(nn.AdaptiveAvgPool2d(1),
  26.                                  nn.Conv2d(ch,ch//re,1),
  27.                                  nn.ReLU(inplace=True),
  28.                                  nn.Conv2d(ch//re,ch,1),
  29.                                  nn.Sigmoid())
  30.         self.sSE = nn.Sequential(nn.Conv2d(ch,ch,1),
  31.                                  nn.Sigmoid())
  32.     def forward(self, x):
  33.         return x * self.cSE(x) + x * self.sSE(x)  #scSE

论文原文：Concurrent Spatial and Channel Squeeze & Excitation in Fullly Convolutional Networks文章源自联网快讯-https://x1995.cn/4534.html

继续阅读
历史上的今天
4月
18
Miracle
  • 本文由 发表于 2019年4月18日17:22:48