involution.pytorch()
一个PyTorch实现使用过密的
对合的非官方pytorch实现。 官方实现可以在找到。
特征
该层可以处理任意输入和输出通道,内核大小,步幅和减速比。 但是,输入通道应按组划分。
要求
pytorch >= 1.4.0
einops >= 0.3.0
用法
一个例子:
>>> import torch
>>> from involution import Involution
>>>
>>> x = torch.rand(2,8,5,5)
>>> i = Involution(in_channels=8, out_channels=4, groups=4, kernel_size=3, stride=2, reduction_ratio=2)
>>> i(x).size ()
torch.Size([2, 4, 3, 3])
去做
Ima
1