2017-02-17 97 views
2

我想implememt多任务神经网络在tensorflow,为此,我需要输入:如何在张量流中给图像添加多个标签?

[图像label1的标签2]

,我可以给你的训练神经网络。

我的问题是,如何将TFRecord文件中的多个标签与图像关联?

我目前使用的是build_image_data.py用于生成TFRecord文件的初始模型文件,但在这种情况下,每个图像只有一个标签。

回答

2

他们总是有两个标签吗?如果是这样,请将“label1-label2”作为一个标签。 或者干脆建立两个网络,一个用于标签1,另一个用于标签2.

它们是分层标签吗?然后,检查分层分类器。

0
  1. 我得到了这个工作。对于查找引用的任何人,您可以修改build_image_data.py文件的Example proto并将其与两个标签相关联。 :)
相关问题