如何使用 Google Colab 在 Keras 上加载单个图像？答案

【问题标题】：How can I load a single image on Keras, using Google Colab?如何使用 Google Colab 在 Keras 上加载单个图像？
【发布时间】：2020-07-15 23:23:22
【问题描述】：

我已经建立了一个对猫和狗进行分类的深度学习模型。我已经成功安装了谷歌驱动器并根据需要训练了图像。但是，我正在尝试通过上传一个图像并让 Keras 进行预测来进行单个预测。

在像 Spyder 这样的常规 IDE 中，它是这样的：

    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() 方法上编写目录，但此时已经没有想法了。

【问题讨论】：

为了详细说明，您从驱动器加载了 colab 上的数据。在 py 文件上训练您的网络。现在您想对单个图像进行预测。对吗？
解决了，我的目录写错了！谢谢

标签： tensorflow image-processing keras deep-learning google-colaboratory

【解决方案1】：

您可以安装驱动器，通过身份验证连接，然后导入您想要的文件并使用您的模型进行预测。请查看 GitHub 要点 here。您可以按照以下步骤操作，如果您需要任何帮助，请告诉我。谢谢！

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])

【讨论】：