YOLO格式数据转coco格式数据

本文链接：https://blog.csdn.net/weixin_43722052/article/details/136558380

import os
import json
import cv2
import random
import time
from PIL import Image

coco_format_save_path = '/home/rendoudou/datasets/yolo2coco-new/json'  # 要生成的标准coco格式标签所在文件夹
yolo_format_classes_path = 'name.txt'  # 类别文件，一行一个类
yolo_format_annotation_path = '/home/rendoudou/datasets/yolo2coco-new/cotton-weed/train/labels'  # yolo格式标签所在文件夹
img_pathDir = '/home/rendoudou/datasets/yolo2coco-new/cotton-weed/train/images'  # 图片所在文件夹

with open(yolo_format_classes_path, 'r') as fr:  # 打开并读取类别文件
    lines1 = fr.readlines()
# print(lines1)
categories = []  # 存储类别的列表
for j, label in enumerate(lines1):
    label = label.strip()
    categories.append({
   
   'id': j + 1, 'name': label, 'supercategory': 'None'})  # 将类别信息添加到categories中
# print(categories)

write_json_context = dict()  # 写入.json文件的大字典
write_json_context['info'] = {
   
   'description': '', 'url': '', 'version': '', 'year':