gpt4 book ai didi

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

转载 作者:太空宇宙 更新时间:2023-11-04 02:56:12 25 4
gpt4 key购买 nike

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

[image label1 label2]

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

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

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

最佳答案

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

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

关于python - 如何在 tensorflow 中为一张图片提供多个标签?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/42290182/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com