Question

我在TensorFlow中使用传输学习。我需要使用Inception V3模型来计算图片的特征向量。我在计算JPG格式图片时的代码没问题，但PNG格式的计算将会是错的。

# read model
with gfile.FastGFile(os.path.join(MODEL_DIR, MODEL_FILE), 'rb') as f:
    graph_def = tf.GraphDef()
    graph_def.ParseFromString(f.read())

bottleneck_tensor, jpeg_data_tensor = tf.import_graph_def(graph_def, return_elements=[BOTTLENECK_TENSOR_NAME, JPEG_DATA_TENSOR_NAME])

......

# get imagepath
image_path = get_image_path(image_lists, INPUT_DATA, index, category)
# read image
image_data = gfile.FastGFile(image_path, 'rb').read()

# calculate the feature vector
# **This statement is wrong when png images**
bottleneck_values = sess.run(bottleneck_tensor, {jpeg_data_tensor: image_data})

控制台错误包括：

...... 

Not a JPEG file: starts with 0x89 0x50

......

InvalidArgumentError (see above for traceback): Invalid JPEG data, size 19839
     [[Node: import/DecodeJpeg = DecodeJpeg[acceptable_fraction=1, channels=3, dct_method="", fancy_upscaling=true, ratio=1, try_recover_truncated=false, _device="/job:localhost/replica:0/task:0/cpu:0"](_recv_import/DecodeJpeg/contents_0)]]

我想错误的关键是阅读代码图片，但我不知道如何修改它以支持PNG格式，有人可以帮助我吗？

由于

Answer 1

你能下载一些其他图片，看看它们是否有效？问题是我认为你的图像加载......

否则，尝试使用pyplot读取图像：

{{1}}

Answer 2

我找到了解决方案。虽然我无法修改它以支持PNG格式，但我可以读取PNG图像并将其转换为JPEG格式。添加代码如下：

oThis

此方法不会在磁盘中生成新图像，这没关系！

如何使用Inception v3模型计算转移学习中PNG图像的特征向量

2 个答案: