Использование resnet50 для классификации изображений - PullRequest
0 голосов
/ 27 марта 2020

Я пытаюсь использовать resnet50 для классификации серии из двух изображений. В методе прогнозирования я получил следующую ошибку:

ValueError: Если ваши данные представлены в виде тензоров symboli c, вы должны указать аргумент steps (вместо batch_size аргумент, поскольку ожидается, что тензоры символьных c будут генерировать пакеты входных данных).

Пока мои тензоры не являются символическими c. Вот мой код:

   flower = load_sample_image('flower.jpg')/ 255

   images = np.array([china,flower])
   model = keras.applications.resnet50.ResNet50(weights='imagenet')
   resized_images = tf.image.resize(images, [224,224])

   inputs = keras.applications.resnet50.preprocess_input(resized_images*255)
   model.predict(inputs)```
...