物体检测分类

数据挖掘 张量流 多标签分类 物体识别 标签 物体检测
2022-02-18 14:45:10

我目前正在使用 TensorFlow 对象检测 API 训练一个用于检测电阻器的分类器。为此,我从 ImageNet 下载了电阻器图像,我目前正在标记将用于训练的那些。

但是,我在训练图像中发现了两种不同类型的电阻器:

常用电阻有4条指示值 上面写有值的电阻器

我想知道将它们放在同一个标​​签下是否安全,或者将它们分类在两个不同的标签下会更好,这样训练会更准确?

1个回答

通常,您希望向您的模型展示它会在现实世界中看到的所有类型的“电阻器”。

您手动浏览数据集并将这些图像中的每一个标记为电阻器 A 或电阻器 B 是否可行?您是否关心您的模型是否可以做出这种区分?如果其中任何一个的答案是否定的,我会将它们全部保留为一个标签并训练模型。您的模型可能没有足够的训练数据来了解这些是不同类型的电阻器,但您可以在训练后进行评估。