文件名称:involution.pytorch:使用einops https的Involution的PyTorch实现
文件大小:69KB
文件格式:ZIP
更新时间:2024-04-14 10:37:31
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)