InvalidArgumentError: глубина ввода должна делиться равномерно на глубину фильтра: 4 против 3 - PullRequest
Новая
       61

InvalidArgumentError: глубина ввода должна делиться равномерно на глубину фильтра: 4 против 3

1 голос
/ 11 февраля 2020

Я новичок. Я попробовал классификацию изображений по Tensorflow и получил следующую ошибку. Я нашел похожую проблему в Интернете, но я не мог понять. Что означает ошибка? Как мне это сделать? Пожалуйста, дайте мне совет. Я использую 100 файлов (png / 15pix, 15pix) как образец изображения. Tensorflow ver.2.0.0 / python ver.3.8.1 / Блокнот Jupyter.

sample image

    num_epochs = 30
    steps_per_epoch = round(num_train)//BATCH_SIZE
    val_steps = 20
    history = model.fit(train_data.repeat(),
                epochs=num_epochs,
                steps_per_epoch = steps_per_epoch,
                validation_data=val_data.repeat(), 
                validation_steps=val_steps)

InvalidArgumentError: глубина ввода должен быть равномерно делим по глубине фильтра: 4 против 3 [[node sequential_2 / mobilenetv2_1.00_96 / Conv1 / Conv2D (определено в C: \ Users \ XXXXX \ Anaconda3 \ lib \ site-packages \ tensorflow_core \ python \ framework \ ops .py: 1751)]] [Op: __inference_distributed_function_42611] Стек вызовов функций: распределенная_функция

1 Ответ

0 голосов
/ 13 февраля 2020

Я нашел ответ! В моем случае это помогла следующая программа. 

XXX = tf.convert_to_tensor(XXX[:,:,:3])

I wi sh это вам тоже поможет. Спасибо.

...