Yolov5的json文件转txt文件中，归一化的处理。

最新推荐文章于 2024-07-06 15:02:46 发布

小白误入大佬村

最新推荐文章于 2024-07-06 15:02:46 发布

阅读量4.8k

点赞数 3

文章标签： python 目标检测

本文链接：https://blog.csdn.net/m0_60781841/article/details/123965654

版权

该博客介绍了如何将JSON文件中的bbox数据进行归一化处理，用于目标检测。两种bbox表示方式——(x1,y1,x2,y2)和(x,y,w,h)——被讨论，并提供了对应的Python代码实现。转换过程涉及计算中心点坐标、宽和高，然后根据图像尺寸进行归一化。

摘要由CSDN通过智能技术生成

1.任务分析

json转txt主要就是处理json文件中的bbox，而目标检测中，bbox有两种储存形式，一种里面储存的是（x1,y1,x2,y2）,另一种是(x,y,w,h)。以下图为例：

在进行转化之前，你需要知道你的json文件里面bbox存储的是哪种形式。

2.归一化代码实现

2.1若bbox（x1,y1,x2,y2）

def convert(size,box):
    dw = 1. / size[0]
    dh = 1. / size[1]
    x = (box[0] + box[2]) / 2.0
    y = (box[1] + box[3]) / 2.0
    w = box[2] - box[0]
    h = box[3] - box[1]
    x = x * dw
    w = w * dw    
    y = y * dh
    h = h * dh
    return(x,y,w,h)

2.2若bbox（x,y,w,h）

    def convert(siez,box):
        x, y, w, h = item['bbox']
        dw = 1. / size[0]
        dh = 1. / size[1]
        x = x * dw
        w = w * dw
        y = y * dh
        h = h * dh
        return(x,y,w,h)