YOLO格式标注数据转COCO标注数据

本文介绍了一个Python脚本,用于将person类别的人脸标注转换为支持多类别信息的JSON格式,包括图片信息、注释区域和类别映射。核心内容涉及读取文本标注文件,处理图像尺寸,并将每个标签信息组织成标准的JSON结构。

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

这里仅仅考虑person类别,如果考虑其他类别,则需要增加类别信息,稍作调整即可。

import json
import os
import imagesize
import copy


def txt_to_json(img_dir,annotation_dir,json_path,img_format='.jpg',annotation_format='.txt'):
    # json 文件主要两项内容
    json_dict = dict()
    annotations = list()
    images = list()
    categories = list()
    # 一个标签和一张图
    one_annotation = dict()
    one_image = dict()
    annotation_bbox_id = 0
    for file in os.listdir(annotation_dir):
        if file.endswith(annotation_format):
            # 读取图片信息:长宽,整合到one_image当中
            one_image[
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值