Как я могу загрузить одно изображение на Keras, используя Google Colab? - PullRequest
Новая
       24

Как я могу загрузить одно изображение на Keras, используя Google Colab?

0 голосов
/ 03 апреля 2020

Я построил модель глубокого обучения, которая классифицирует кошек и собак. Я успешно подключил Google диск и обучил изображения по мере необходимости. Однако я пытаюсь сделать прогноз single , загрузив изображение one и сделав прогноз Keras.

В обычной IDE, такой как Spyder, это выглядит так: 

    test_image = image.load_img('image1.jpg',target_size=(64,64))

Но выдает эту ошибку: 

   Transport endpoint is not connected: 'image1.jpg'

Я перемонтировал диск, а затем он говорит мне: 

    No such file or directory: 'image1.jpg'

После этого я поиграл с как бы я написал каталог для метода image.load (), но на этом этапе у меня не осталось идей.

1 Ответ

0 голосов
/ 05 апреля 2020

Вы можете смонтировать диск, подключиться с аутентификацией, затем импортировать нужные файлы и прогнозировать, используя вашу модель. Пожалуйста, проверьте GitHub Gist здесь . Вы можете выполнить следующие действия и сообщить мне, если вам нужна помощь. Спасибо! 

from google.colab import drive
# This will prompt for authorization.
drive.mount('/content/drive')

# Predicting Roses
img=mpimg.imread('/content/drive/My Drive/5602220566_5cdde8fa6c_n.jpg')
imgplot = plt.imshow(img)
img = tf.expand_dims(img,0) # need this to make batch_shape = 1
img=img/255  # normalizing the image
img=tf.image.resize(img,size=(224, 224)) # resizing image
Prob=loaded_model.predict(img) # prediction
indd=tf.argmax(Prob[0],axis=-1).numpy()
print(tf.argmax(Prob[0],axis=-1).numpy())
print(labels_string[indd])
...