Question

我正在尝试运行注意力模型，但是当我尝试导入spatial_transformer时，它说没有名为“ spatial_transformer”的模块，因此我尝试使用“ pip install space_transformer”，但结果是“没有匹配的分布”发现于spacetransformer'。我搜索了这个问题，看到其他人使用相同的代码“ from space_transformer import SpatialTransformer”，所以我想知道如何安装此软件包，谢谢！

PS：我的代码如下：

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

这还可能意味着从文件space_transformer.py导入了“ SpatialTransformer”。将其放置在与python脚本相同的目录中应该可以工作。

例如取这个https://gist.github.com/oarriaga/8438a7276bdc2b4ff03986d465e243b3并将其保存为space_transformer.py。

如何安装spatial_transformer

2 个答案: