如何从Keras加载本地图像?

Question

我试图在Keras中加载本地图像，但它不工作，当试图显示它时，我得到一个错误:

/usr/local/lib/python3.7/dist-packages/IPython/core/formatters.py:364: FormatterWarning: image/png formatter returned invalid type <class 'PIL.JpegImagePlugin.JpegImageFile'> (expected (<class 'bytes'>, <class 'str'>)) for object: <IPython.core.display.Image object>
FormatterWarning
<IPython.core.display.Image object>

我正在使用Keras创始人创建的这个笔记本(它自己工作，但它从互联网下载了一个图像):https://colab.research.google.com/drive/1H-lNMxVF9NPME6oZVphFRoi7yglJMZxP

我想尝试用我自己的图像来玩风格转换，所以我改变了这个:

import numpy as np
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras.applications import vgg19
# base_image_path = keras.utils.get_file('my_pic.jpg', './my_pic.png')
base_image_path = keras.utils.load_img('/content/my_pic.jpg')
style_reference_image_path = keras.utils.get_file('starry_night.jpg', 'https://i.imgur.com/9ooB60I.jpg')
result_prefix = 'paris_generated'
iterations = 5000
# Weights of the different loss components
total_variation_weight = 1e-6
style_weight = 2e-6
content_weight = 2e-8
# Dimensions of the generated picture.
width, height = keras.preprocessing.image.load_img(base_image_path).size
img_nrows = 400
img_ncols = int(width * img_nrows / height)

当我做这个修改时:

# base_image_path = keras.utils.get_file('my_pic.jpg', './my_pic.png')
base_image_path = keras.utils.load_img('/content/my_pic.jpg')

所有代码停止工作。我认为get_file方法正在做一些事情，使其可访问作为一个图片，因为在原始的笔记本中，它显示图片，但对于我的，它没有。

有什么我能做的吗?我是全新的图像工作，但我做了研究，我认为get_file方法是通过网络获取文件，所以我认为它不会工作，如果我复制一个文件在我的本地协作系统。

Answer 1

我个人会阅读很多tensorflow的文档。对于大多数函数和你想做的事情来说，它都很神奇。例如get_file你可以看到它说返回Path to the downloaded file。而load_img将使用路径将文件加载到PIL图像对象中。

在这个例子中，他们正在做的是从互联网下载文件get_file返回下载文件的路径，然后他们使用keras.preprocessing.image.load_img加载图像并获得尺寸(即宽度，高度)

所以解决你的问题。很容易。不要给keras.utils.load_img('/content/my_pic.jpg')打电话。你只需要设置

base_image_path = '/content/mypic.jpg'

请记住，这个文件路径意味着您在驱动器的根目录下有一个名为content的文件夹。如果您打算放置一个相对路径，那么删除第一个/。

例如如果在colab/你的实验室有一个名为content的文件夹在你的笔记本旁边，那么你想使用相对路径content/mypic.jpg。但是，如果映像位于您的下载中，那么像这样做作为绝对路径(例如C:UsersUSERDownloads或/Users/USER/downloads)