keras 如何安装空间变换器

dgiusagp  于 2023-04-06  发布在  其他
关注(0)|答案(3)|浏览(107)

我试图运行一个注意力模型,但当我尝试导入spatial_transformer时,它说没有名为'spatial_transformer'的模块,所以我尝试使用'pip install spatial_transformer',但结果是'No matching distribution found for spatialtransformer'。我搜索这个问题,看到其他人使用相同的代码“from spatial_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
4sup72z8

4sup72z81#

使用pytorch实现代替。
这是一个空间转换器,有清晰的文档和示例。
https://pytorch.org/tutorials/intermediate/spatial_transformer_tutorial.html
我们没有太多关于你在这里导入的模块的信息,因为包名不是通用的。

toiithl6

toiithl62#

关于这条线

from spatial_transformer import SpatialTransformer

也可以从spatial_transformer. py文件中导入'SpatialTransformer'。将其放在与python脚本相同的目录中应该可以工作。
例如,获取此https://gist.github.com/oarriaga/8438a7276bdc2b4ff03986d465e243b3并将其保存为spatial_transformer. py。

dxpyg8gm

dxpyg8gm3#

可以使用以下命令安装spatial_transformer

pip install stn

参见PyPI链接

相关问题