как использовать модель Keras для предсказания изображения?

Я выполнил поездку и получил модель в формате.hdf5

Нейронная сеть, которую я использую, - это сиамская сверточная нейронная сеть.

при проверке предсказанное изображение является случайным изображением из моей тестовой папки. я использую это при тестировании

test_alphabets = glob('{}/TEST/*'.format(dataset_dirname))

testset={}
for alph in test_alphabets:
    dirs = glob('{}/*'.format(alph))
    alphabet = {}
    for dirname in dirs:
        alphabet[dirname] = glob('{}/*'.format(dirname))
    testset[alph] = alphabet        

затем отобразите результат с помощью

display_validation_test(siamese_model1, testset)

результат выглядит следующим образом. Как выполнить тестовый процесс, введя нужное изображение, а затем отобразив соответствующее изображение, используя ранее использованную модель.h5?

1 ответ

  1. Сначала вы создаете свою модель (экземпляр keras.Model или keras.Sequential) с той же архитектурой, что и модель, которую вы обучили.
  2. загрузить веса из файла.h5 model.load_weights('your_weight_file.h5')
  3. прочтите ваше изображение (я). Если изображение одно, обязательно добавьте 1 в качестве размера партии.
  4. Вызов предсказать:prediction = model.predict(images)
Другие вопросы по тегам