我如何在tensorlfow中输入图像功能？

Question

我有一个与张量流有关的问题。

我使用了DNNClassifier模型。

我想要图像功能，确定存在许多其他列 （例如，年龄，类型..）

在年龄的情况下，我使用tf.feature_column.numeric_column（年龄）。

但是，我不知道输入图像功能。

请帮助我......谢谢。

Answer 1

图像处理本质上是数字处理，因此您可以使用tf.feature_column.numeric_column作为像素。只需确保将shape参数设置为图像的大小：

image_col = tf.feature_column.numeric_column('pixels', shape=[image_width * image_height])

This article提供了使用DNNClassifier对图像进行分类的更全面的讨论。