COCO2YOLO

最新推荐文章于 2025-01-13 17:37:47 发布

Leagues

最新推荐文章于 2025-01-13 17:37:47 发布

阅读量573

点赞数 1

文章标签： python

本文链接：https://blog.youkuaiyun.com/Leagues/article/details/131071420

版权

该Python脚本用于将包含形状信息的COCO格式标注数据转换为YOLO数据格式。它读取源路径中的.json文件，提取类别和边界框信息，然后按YOLO的格式要求写入目标路径的.txt文件中，包括类索引、中心点坐标和宽高相对于图像的比例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Label打标数据转成YOLO数据格式

# from ultralytics.yolo.data.converter import convert_coco
#
# convert_coco(labels_dir=r'D:\github\ultralytics\111')
import cv2
import json
import os

source_path = r"D:\github\ultralytics\111"

dest_path = r"D:\github\ultralytics\222"
class_list = [ 'cut_corner' ,'with_corner']


for jsonfile in os.listdir(source_path):
    if jsonfile.endswith(".json"):
        with open(source_path + '//' + jsonfile, 'r') as f:    #打开json文件，读取数据
            data = json.load(f)
            #
            class_name = data.get("shapes")[0].get("label")

            points = data.get("shapes")[0].get("points")

            imageHeight = data.get("imageHeight")
            imageWidth = data.get("imageWidth")

            Center_X = float(points[0][0] + points[1][0]) /2
            Center_Y = float(points[0][1] + points[1][1]) /2

            Width = abs(float(points[0][0]) - float(points[1][0]))
            Height = abs(float(points[0][1]) - float(points[1][1]))

            #算出类别、四个点的坐标
            #新建一个txt同名文件
            file = open(dest_path + "\\"+ jsonfile.split(".json")[0] + '.txt', 'a')
            file.write(str(class_list.index(class_name)) + ' ' + str(Center_X / imageWidth) + ' ' + str(Center_Y / imageHeight) + ' ' + str(Width / imageWidth) + ' ' + str(Height / imageHeight))