上传者: 38608875
|
上传时间: 2021-12-22 19:20:13
|
文件大小: 270KB
|
文件类型: -
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__