标签工作室中的预注释(yolo v5) 问题



我正试图在label-studio中使用我的Yolo v5模型的注释。我可以把注释带到label-studio并展示它们。问题是label-studio显示的标签不准确(我有从Yolo生成的预测应该是什么样子的图像)。我对从yolo获得的x, y,宽度,高度数据所做的唯一转换是乘以100。

yolo的预测:YOLOlabel-studio:label-studio

请使用label-studio-converter: https://github.com/heartexlabs/label-studio-converter/pull/46

很可能你计算错了坐标,正确的代码在这里:https://github.com/heartexlabs/label-studio-converter/blob/master/label_studio_converter/imports/yolo.py#L85

我对从yolo获得的x, y,宽度,高度数据所做的唯一转换是乘以100

在@Max答案的基础上,您可以使用以下公式(从这里)将yolo坐标转换为标签工作室兼容的坐标

x_lbstdio = (x-width/2)*100
y_lbstdio = (y-height/2)*100
w_lbstdio = w*100
h_lbstdio = h*100

请注意,唯一的变化是分别从x和y中减去宽度和高度的一半,然后乘以100

最新更新