machine-learning - 如何将边界框 (x1, y1, x2, y2) 转换为 YOLO 样式 (X, Y, W, H)

Question

我正在训练 YOLO 模型，我有以下格式的边界框:-

x1, y1, x2, y2 => ex (100, 100, 200, 200)

我需要将其转换为 YOLO 格式，如下所示:-

X, Y, W, H => 0.436262 0.474010 0.383663 0.178218

我已经计算出了中心点X、Y、高度H和重量W。但仍然需要将它们转换为如上所述的 float 。

Answer 1

对于那些寻找问题相反的人(yolo 格式到普通 bbox 格式)

def yolobbox2bbox(x,y,w,h):
    x1, y1 = x-w/2, y-h/2
    x2, y2 = x+w/2, y+h/2
    return x1, y1, x2, y2