Есть ли способ назначить входные данные модели Python Tensorflow Keras для переменной? - PullRequest
Новая
       22

Есть ли способ назначить входные данные модели Python Tensorflow Keras для переменной?

0 голосов
/ 28 марта 2019

Я строю модель Keras, и я определил класс, содержащий ее. Мне нужно "сохранить" вход модели в нечто вроде self.stored_input = input_of_the_model

Я использую Python 3.6.7, Tensorflow 2.1.6 и Keras 2.2.4 Я пробовал предварительно, но без хороших результатов. Вот что я делаю:

'' '

Класс АвтоЭнкодер: def init (self, input_shape = [224, 224]): 

    self.models_dir = '../model'


    # Input
    img = Input(shape=(224, 224, 1))

    ..... Convolutional neural network layers here.....

    self.img = img

'' '

Я хочу использовать self.img в функции класса AutoEncoder, но я получаю эту ошибку при ее использовании:

InvalidArgumentError: Вы должны передать значение для тензора-заполнителя 'input_1' с плавающей точкой dtype и формой [?, 224,224,1] [[{{node input_1}} = Placeholderdtype = DT_FLOAT, shape = [?, 224,224,1], _device = "/ job: localhost / replica: 0 / task: 0 / device: CPU: 0"]]

...