位置共享卷积

网友投稿 978 2022-05-29

Involution的一个特点:输入通道和输出通道必须是一样的,如果想实现输入和输出不一样,文末有方法。

实现1:

https://github.com/d-li14/involution/blob/21c3158fcbb4ecda8ed4626fcae8b01be511a598/cls/mmcls/models/utils/involution_naive.py

以来mmcv

mmcv安装:

pip install mmcv

import torch.nn as nn

from mmcv.cnn import ConvModule

class involution(nn.Module):

def __init__(self,

channels,

kernel_size,

stride):

super(involution, self).__init__()

self.kernel_size = kernel_size

self.stride = stride

self.channels = channels

reduction_ratio = 4

位置不共享卷积

self.group_channels = 16

self.groups = self.channels // self.group_channels

self.conv1 = Con

神经网络

版权声明:本文内容由网络用户投稿,版权归原作者所有,本站不拥有其著作权,亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容,请联系我们jiasou666@gmail.com 处理,核实后本网站将在24小时内删除侵权内容。

上一篇:smbexec的使用(c$共享)
下一篇:如何实现团队高效协作办公?
相关文章