我怎么能在python中加载自己的图像到网络?

时间:2018-03-19 13:37:27

标签: python-3.x tensorflow conv-neural-network

我已经创建了一个卷积神经网络来创建数据。现在我想将输入更改为我的图像。我该怎么做?需要以特定的格式保存图片吗?另外,如何保存所有图片并一个接一个地训练?我在tensorflow中使用python。

1 个答案:

答案 0 :(得分:1)

Tensorflow支持开箱即用的bmp,​​gif,jpeg和png。

因此加载数据(将文件作为字符串类型的0D张量读入内存),然后将其传递给tf.image.decode_image或其中一个专用函数,如果由于某种原因它不起作用。

您应该将图像作为形状[宽度,高度,通道]的张量返回(如果您只有单个通道图像,如灰度,则可能缺少通道。)

为了使这项工作更好,你应该让所有的图像都采用相同的格式。如果您可以将所有图像加载到ram中并批量传递它,那么它可能是最简单的事情。下一个最简单的方法是将图像复制到tensorflow.Example并复制到tf.TFRecordReader以进行混洗和批处理。如果所有其他方法都失败了,我认为您可以设置输入函数来按需读取图像并通过批处理机制管理它们,但我不确定如何做到这一点。

这里是与图像相关的tensorflow文档的link