如何为YOLO标注训练数据

我想问一下如何为YOLO算法标注训练数据。

假设每个标签Y,我们需要指定[Pc, bx, by, bh, bw],其中Pc表示对象是否存在(1=存在,0=不存在),(bx, by)是目标对象中心的相对位置,(bh, bw)是包含目标对象的边界框的相对尺寸。

以下的图片作为例子,包含黑色汽车的单元格(1,2)应该有标签Y = [1, 0.4, 0.3, 0.9, 0.5]。而对于没有汽车的单元格,它们应该有标签[0, ?, ?, ?, ?] [Coursera Deep Learning Specialization Materials]1

但如果我们有一个更细的网格,像这样,每个单元格的尺寸小于真实边界框。enter image description here

假设汽车的真实边界框是红色的框,真实中心点是红色的点,位于单元格2中。

对于单元格2,它的标签Y = [1, 0.9, 0.1, 2, 2],这是正确的吗?对于单元格1、3、4,它们会有什么样的标签?它们会有Pc=1还是Pc=0?如果Pc=1,bx和by会是多少?(我记得bx, by的值应该在0和1之间。但在单元格1、3、4中,没有目标对象的中心点)


回答:

实际上,中点位于单元格2中。根据YOLO算法,单元格1、3、4的Pc=0,因为该算法只考虑包含中点的单元格,并计算边界框,如您所述,使用bx, by, bh, bw。对于Y = [1, 0.9, 0.1, 2, 2]的建议,我认为如果您将(0,0)点作为左上角,Y更可能是Y = [1,0.2,0.57,0.21,0.15],这是基于下面的19 x 19网格计算得出的。

enter image description here

Related Posts

在使用k近邻算法时,有没有办法获取被使用的“邻居”?

我想找到一种方法来确定在我的knn算法中实际使用了哪些…

Theano在Google Colab上无法启用GPU支持

我在尝试使用Theano库训练一个模型。由于我的电脑内…

准确性评分似乎有误

这里是代码: from sklearn.metrics…

Keras Functional API: “错误检查输入时:期望input_1具有4个维度,但得到形状为(X, Y)的数组”

我在尝试使用Keras的fit_generator来训…

如何使用sklearn.datasets.make_classification在指定范围内生成合成数据?

我想为分类问题创建合成数据。我使用了sklearn.d…

如何处理预测时不在训练集中的标签

已关闭。 此问题与编程或软件开发无关。目前不接受回答。…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注