Pytorch实现各种2d卷积示例
作者:Learn2Learn 时间:2023-06-20 16:17:29
普通卷积
使用nn.Conv2d(),一般还会接上BN和ReLu
参数量NNCin*Cout+Cout(如果有bias,相对来说表示对参数量影响很小,所以后面不考虑)
class ConvBNReLU(nn.Module):
def __init__(self, C_in, C_out, kernel_size, stride, padding, affine=True):
super(ConvBNReLU, self).__init__()
self.op = nn.Sequential(
nn.Conv2d(C_in, C_out, kernel_size, stride=stride, padding=padding, bias=False),
nn.BatchNorm2d(C_out, eps=1e-3, affine=affine),
nn.ReLU(inplace=False)
)
def forward(self, x):
return self.op(x)
深度可分离卷积depthwise separable convolution
卷积操作可以分为NN 的Depthwise卷积(不改变通道数)和11的Pointwise卷积(改变为输出通道数),同样后接BN,ReLU。参数量明显减少
参数量:
NNCin+Cin11*Cout
class SepConv(nn.Module):
def __init__(self, C_in, C_out, kernel_size, stride, padding, affine=True):
super(SepConv, self).__init__()
self.op = nn.Sequential(
nn.ReLU(inplace=False),
nn.Conv2d(C_in, C_in, kernel_size=kernel_size, stride=stride, padding=padding, groups=C_in, bias=False),
nn.Conv2d(C_in, C_out, kernel_size=1, padding=0, bias=False),
nn.BatchNorm2d(C_out, eps=1e-3, affine=affine)
)
def forward(self, x):
return self.op(x)
空洞卷积dilated convolution
空洞卷积(dilated convolution)是针对图像语义分割问题中下采样会降低图像分辨率、丢失信息而提出的一种卷积思路。利用添加空洞扩大感受野。
参数量不变,但感受野增大(可结合深度可分离卷积实现)
class DilConv(nn.Module):
def __init__(self, C_in, C_out, kernel_size, stride, padding, dilation, affine=True):
super(DilConv, self).__init__()
self.op = nn.Sequential(
nn.ReLU(inplace=False),
nn.Conv2d(C_in, C_in, kernel_size=kernel_size, stride=stride, padding=padding, dilation=dilation, groups=C_in, bias=False),
nn.Conv2d(C_in, C_out, kernel_size=1, padding=0, bias=False),
nn.BatchNorm2d(C_out, eps=1e-3, affine=affine),
)
def forward(self, x):
return self.op(x)
Identity
这个其实不算卷积操作,但是在实现跨层传递捷径
class Identity(nn.Module):
def __init__(self):
super(Identity, self).__init__()
def forward(self, x):
return x
来源:https://blog.csdn.net/u014767662/article/details/88574643
标签:Pytorch,2d,卷积
0
投稿
猜你喜欢
Python打包模块wheel的使用方法与将python包发布到PyPI的方法详解
2022-03-26 10:52:57
Python+eval函数实现动态地计算数学表达式详解
2023-08-13 22:39:11
Python多线程中阻塞(join)与锁(Lock)使用误区解析
2022-03-22 08:00:31
python面试题之列表声明实例分析
2022-01-10 12:46:51
python保存字典和读取字典的实例代码
2023-05-12 14:17:54
python openssl模块安装及用法
2023-07-03 12:01:04
Python生成九宫格图片的示例代码
2023-03-16 20:29:38
python 使用tkinter+you-get实现视频下载器
2023-08-15 18:40:25
在Python中操作列表之List.append()方法的使用
2021-07-07 04:09:45
Python 2/3下处理cjk编码的zip文件的方法
2022-08-05 17:53:40
python代码过长的换行方法
2022-12-25 07:19:59
python 爬虫 实现增量去重和定时爬取实例
2022-01-26 01:01:37
pycharm使用anaconda全过程
2023-07-19 04:57:12
详解用python实现爬取CSDN热门评论URL并存入redis
2022-08-30 00:11:35
python opencv图片编码为h264文件的实例
2023-01-12 13:22:07
PHP开发实现快递查询功能详解
2023-11-24 12:19:39
php获取通过http协议post提交过来xml数据及解析xml
2023-11-14 15:43:36
PyCharm第一次安装及使用教程
2022-06-21 23:18:00
Python filter()及reduce()函数使用方法解析
2022-09-05 07:01:59
Python中的 if 语句及使用方法
2022-12-19 16:35:10