如何安装spatial_transformer

Question

我正在尝试运行一个注意力模型，但是当我尝试导入spatial_transformer时，它显示没有名为'spatial_transformer‘的模块，所以我尝试使用'pip install spatial_transformer'，但结果是’没有为spatial_transformer找到匹配的分布‘。我搜索这个问题，看到其他人使用相同的代码“从spatial_transformer导入SpatialTransformer"，所以我想知道如何安装这个包，谢谢！

附言:我的代码如下：

import numpy as np
import keras
import keras.backend as K
import matplotlib.pyplot as plt
from scipy.misc import imresize
from keras.datasets import mnist
from keras.models import Sequential
from keras.layers.core import Dense, Dropout, Activation, Flatten
from keras.layers.convolutional import Convolution2D, MaxPooling2D, 
Conv2D
from keras.layers import Input
from keras.utils import np_utils, generic_utils
from keras.optimizers import Adam, SGD
from spatial_transformer import SpatialTransformer
#from stn.transformer import spatial_transformer_network
%matplotlib inline

np.random.seed(1337)  # for reproducibility
batch_size = 128
nb_classes = 10
nb_epoch = 12

Answer 1

请改用pytorch实现。

它是一个空间转换器，有清晰的文档和示例。

https://pytorch.org/tutorials/intermediate/spatial_transformer_tutorial.html

我们没有太多关于您在这里导入的模块的信息，因为包名称并不通用。

Answer 2

关于这条线，

从spatial_transformer导入 SpatialTransformer

这也可能意味着从文件spatial_transformer.py导入'SpatialTransformer‘。将它放在与您的python脚本相同的目录中应该可以工作。

例如，获取此https://gist.github.com/oarriaga/8438a7276bdc2b4ff03986d465e243b3并将其保存为spatial_transformer.py。