我如何在 python 中将自己的图像加载到网络？

Question

我已经为 mnist 数据制作了一个卷积神经网络。现在我想将输入更改为我的图像。我该怎么做？需要把图片保存成特定格式吗？另外，如何保存所有图片，一张一张地训练？我在tensorflow和python中使用。

Answer 1

Tensorflow 开箱即可支持 bmp、gif、jpeg 和 png。

因此，加载数据（将文件作为字符串类型的 0D 张量读入内存）然后将其传递给 tf.image.decode_image 或如果由于某种原因无法正常工作的专用函数之一。

您应该将图像作为形状 [width, height, channels] 的张量取回（如果您只有一个通道图像，例如灰度，则通道可能会丢失）。

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

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