【文件属性】:
文件名称:involution.pytorch:使用einops https的Involution的PyTorch实现
文件大小:69KB
文件格式:ZIP
更新时间:2021-03-25 16:50:51
computer-vision deep-learning convolutional-neural-networks Python
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
【文件预览】:
involution.pytorch-main
----images()
--------invo.png(75KB)
----involution.py(2KB)
----README.md(1KB)