как установить пространственный_трансформер

Question

Я пытаюсь запустить модель внимания, но когда я пытаюсь импортировать пространственный_трансформатор, он говорит, что нет модуля с именем «пространственный_трансформатор», поэтому я пытаюсь использовать «pip installatial_transformer», но получается, что «нет соответствующего распределениянайдено для пространтрансформатора.Я ищу эту проблему и вижу, что кто-то другой использует тот же код "из пространственного импорта 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

Относительно этой строки,

из import пространственного_трансформатора SpatialTransformer

это также может означать импорт 'SpatialTransformer' из файла пространственный_трансформатор.py.Поместите его в ту же директорию, что и ваш скрипт на python.

например, взяв это https://gist.github.com/oarriaga/8438a7276bdc2b4ff03986d465e243b3 и сохранив его как пространственный_трансформер.py.

Answer 2

Используйте взамен реализацию pytorch.

Это пространственный преобразователь с понятной документацией и примерами.

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

У нас мало информации омодуль, который вы импортируете сюда, так как имена пакетов не универсальны.