Box坐标转换

文章目录[隐藏]

[x, y, width, height]--在我们的训练数据集中就是这种情况(也叫COCO格式)。
[x1, y1, x2, y2] - 我们在BBox增强阶段创建的格式化版本,也叫[xmin, ymin, xmax, ymax]。这个格式在SSD/RCNN/Fast RCNN/Faster RCNN模型中使用。
[x_center, y_center, width, height] - 这是YOLO格式,或者说是使用YOLO模型训练时使用的格式。x_center, y_center是边界框中心的标准化坐标,width, height是图像的标准化宽度和高度。

Example1:

annotations 数据格式:[{'x': 641, 'y': 634, 'width': 74, 'height': 71}]

变成coco_bbox [[641, 634, 74, 71]]

数据处理:

train_df["coco_bbox"] = train_df["annotations"].apply(lambda annot: [list(item.values()) for item in eval(annot)])

版权声明:本文为CSDN博主「苏坡爱豆的笑容都没你的甜」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/weixin_44656422/article/details/122961815

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐

CVPR2021: Sparse R-CNN新的目标检测模型

今天我们将讨论由四个机构的研究人员提出的一种方法,其中一个是字节跳动人工智能实验室。他们为我们提供了一种新的方法,称为稀疏R-CNN(不要与 Sparse R-CNN 混淆,后者在 3D 计算机视觉任务

实战深度学习目标检测:RCNN (1)

深度学习目标检测:RCNN 什么是目标检测?目标检测主要是明确从图中看到了什么物体?他们在什么位置。传统的目标检测方法一般分为三个阶段:首先在给定的图像上选择一些候选的区域&#xff0c