Skip to main content
 首页 » 编程设计

python之如何在 tensorflow 中为一张图片提供多个标签

2024年10月01日31pengyingh

我想在 tensorflow 中实现多任务神经网络,为此我需要输入:

[image label1 label2]

我可以将其提供给神经网络进行训练。

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

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

请您参考如下方法:

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

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