用pytorch写深度可分离卷积
时间: 2023-03-28 08:00:48 浏览: 113
MixConv:混合感受野的深度可分离卷积(Pytorch实现)
5星 · 资源好评率100%
可以使用PyTorch的nn模块来实现深度可分离卷积。首先,使用nn.Conv2d来实现深度卷积,然后使用nn.Conv2d来实现逐通道卷积。最后,将这两个卷积层组合在一起,就可以得到深度可分离卷积。具体实现方法可以参考PyTorch官方文档或者相关教程。
阅读全文
MixConv:混合感受野的深度可分离卷积(Pytorch实现)