如何在张量流中为图像提供多个标签？

Question

我想在tensorflow中实现多任务神经网络，我需要输入：

  

[image label1 label2]

我可以给神经网络进行训练。

我的问题是，如何在TFRecord文件中将多个标签与图像相关联？

我目前正在使用初始模型的 build_image_data.py 文件来生成TFRecord文件，但在这种情况下，每张图片只有一个标签。

Answer 1

他们总是有两个标签吗？如果是这样，请尝试将“label1-label2”作为一个标签。 或者简单地构建两个网络，一个用于标签1，另一个用于标签2.

他们是分层标签吗？然后，查看Hierarchical classifiers。

Answer 2

1. 我得到了这个工作。对于任何寻找引用的人，您可以修改build_image_data.py文件的示例proto并将其与两个标签相关联。 ：）