文件名称：N-R法matlab代码-RotEqNet:PyTorch/Python的旋转等变网络
文件大小：270KB
文件格式：ZIP
更新时间：2024-06-09 22:22:37
系统开源
NR法matlab代码PyTorch的旋转等变矢量场网络（RotEqNet） 这是在以下方法中提出的方法的PyTorch实现：旋转等变矢量场网络，ICCV 2017，迭戈·马科斯（Diego Marcos），米歇尔·沃尔皮（Michele Volpi），尼科斯·科莫塔基斯（Nikos Komodakis），德维斯·图亚（Devis Tuia）。 原始的MATLAB实现可在以下位置找到： 该代码的目的是提供本文提出的新网络层的实现。 另外，我们尝试重现MNIST-rot数据集的结果以验证实现。 用法示例 from __future__ import division from layers_2D import RotConv , VectorMaxPool , VectorBatchNorm , Vector2Magnitude , VectorUpsampling from torch import nn class MnistNet ( nn . Module ): def __init__ ( self ): super ( MnistNet , self ). __init__
【文件预览】：
RotEqNet-master
----utils.py(8KB)
----__init__.py(0B)
----README.md(3KB)
----mnist()
--------mnist_test.py(7KB)
--------make_mnist_rot.py(2KB)
--------best_model.pt(412KB)
--------download_mnist.py(3KB)
--------mnist.py(3KB)
----layers_2D.py(9KB)