我想在 tensorflow 中实现多任务神经网络,为此我需要输入:
[image label1 label2]
我可以将其提供给神经网络进行训练。
我的问题是,如何将多个标签与 TFRecord 文件中的图像相关联?
我目前正在使用初始模型的 build_image_data.py 文件来生成 TFRecord 文件,但在那种情况下,每张图片只有一个标签。
最佳答案
他们总是有两个标签吗?如果是这样,请尝试将“label1-label2”作为一个标签。 或者简单地构建两个网络,一个用于标签 1,另一个用于标签 2。
它们是分层标签吗?然后,查看分层分类器。
关于python - 如何在 tensorflow 中为一张图片提供多个标签?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/42290182/