四、目标检测txt标注文件（Yolo格式）转为json标注文件（coco格式）

原创

已于 2023-01-06 22:18:12 修改 · 3.5k 阅读

27 ·

CC 4.0 BY-SA版权

文章标签：

#json #python

于 2022-08-22 11:09:03 首次发布

目标检测txt标注文件（Yolo格式）转为json标注文件（coco格式）

读取文件时，不需要设定文件名长度。

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

coco_format_save_path='path'    #要生成的标准coco格式标签所在文件夹
yolo_format_classes_path='path'     #类别文件，一行一个类
yolo_format_annotation_path='path'        #yolo格式标签所在文件夹
img_pathDir='path'                        #图片所在文件夹

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':