Неверный аргумент: Вы должны передать значение для тензора-заполнителя 'conv2d_input' с типом плавания dtype и формой [?, 512,512,3] - PullRequest
Новая
       10

Неверный аргумент: Вы должны передать значение для тензора-заполнителя 'conv2d_input' с типом плавания dtype и формой [?, 512,512,3]

0 голосов
/ 19 марта 2020

Я хочу обучить мою модель, используя эту, но я получил ошибку, это мой код: 


def create_model():
    model = Sequential([
        Conv2D(kernel_size=(3, 3), filters=64, padding='same',
               use_bias=True, activation='relu', input_shape=[512, 512, 3]),
        Conv2D(kernel_size=(3, 3), filters=32, padding='same',
               use_bias=True, activation='relu'),
        Conv2D(kernel_size=(3, 3), filters=1, padding='same',
               use_bias=True, activation='linear')])

    model.compile(optimizer=adam, loss='mse', metrics=['mse'])
    model.summary()
    return model

# GENERATE THE DATASET
bs = 2

def img_train(path):
    for i in os.listdir(path):
        cv2.imread(path + '/' + i)

train_path = img_train('/sresize')

# SAVE THE WEIGHT
model = create_model()
model_checkpoint = tf.keras.callbacks.ModelCheckpoint('Weight{epoch:02d}.h5', save_best_only=True)

history = model.fit(train_path,
                    batch_size=bs,
                    steps_per_epoch=10,
                    epochs=3,
                    callbacks=[model_checkpoint],
                    verbose=1,
                    shuffle=False)

после того, как я ее запустил, я получил это (в заголовке) сообщение об ошибке, кто-нибудь знает как это решить? заранее благодарю за вашу драгоценную помощь.

...