常见的目标检测bbox标注格式

最新推荐文章于 2025-01-21 15:24:58 发布

July_zh

最新推荐文章于 2025-01-21 15:24:58 发布

阅读量673

点赞数 2

文章标签：目标检测人工智能

本文链接：https://blog.csdn.net/chaokudeztt/article/details/137646570

版权

Pascal VOC
bbox：[x_min, y_min, x_max, y_max]
格式：左上右下

COCO
bbox：[x_min, ymin, width, height]
格式：左上宽高

YOLO
bbox [x_center, y_center, width, height]
并进行数据规范化(normalized)
格式：中心坐标，宽高

YOLO转COCO

def xywhn2xyxy(x, w=640, h=640, padw=0, padh=0):
    y = x.clone() if isinstance(x, torch.Tensor) else np.copy(x)
    y[:, 0] = w * (x[:, 0] - x[:, 2] / 2) + padw  # top left x
    y[:, 1] = h * (x[:, 1] - x[:, 3] / 2) + padh  # top left y
    y[:, 2] = w * (x[:, 0] + x[:, 2] / 2) + padw  # bottom right x
    y[:, 3] = h * (x[:, 1] + x[:, 3] / 2) + padh  # bottom right y
    return y

COCO 转 YOLO

 def convert_box(size, box):
        # Convert COCO box to YOLO xywh box
        dw = 1. / size[0]
        dh = 1. / size[1]

        return (box[0] + box[2] / 2) * dw, (box[1] + box[3] / 2) * dh, box[2] * dw, box[3] * dh

Pasic VOC 转 YOLO

def convert_box(size, box):
        # Convert VOC box to YOLO xywh box
        dw = 1. / size[0]
        dh = 1. / size[1]

        return ((box[0] + box[1]) / 2.0 * dw, (box[2] + box[3]) / 2.0 * dh , (box[1] - box[0]) * dw, (box[3] - box[2]) * * dh)