yolo数据集格式(txt)转coco格式,方便mmyolo转标签格式

近期使用mmyolo过程中发现工具自带的yolo2coco.py在转换完数据集格式后,可视化标签的时候会有标签错乱情况,具体原因也没找到,肯定是转换过程代码有问题,于是重新做一份代码直接从yolo数据集转化为coco的json格式。

代码如下:

只需要维护

category_map

data["categories"]

这两个标签id和标签名即可。

我的txt标签格式如下

标签类别为:

{"id": 0, "name": "person"},

{"id": 1, "name": "robotic-arm"}

python 复制代码
import os
import json
from PIL import Image

# 定义类别映射
category_map = {
    "0": 0,
    "1": 1
}

def txt_to_json(input_dir, output_json):
    # 初始化输出数据结构
    data = {
        "images": [],
        "annotations": [],
        "categories": [
            {"id": 0, "name": "person"},
            {"id": 1, "name": "robotic-arm"}
        ]
    }

    # 用于生成唯一的 image_id 和 annotation_id
    image_id = 1
    annotation_id = 1

    # 动态读取图像尺寸
    def get_image_size(image_path):
        with Image.open(image_path) as img:
            return img.width, img.height

    # 遍历输入目录中的所有文件
    for filename in os.listdir(input_dir):
        if filename.endswith('.txt'):
            # 提取图像文件名
            image_name = filename.replace('.txt', '.jpg')
            image_path = os.path.join(input_dir, image_name)

            # 动态读取图像尺寸
            if os.path.exists(image_path):
                image_width, image_height = get_image_size(image_path)
            else:
                print(f"警告: 图像文件 {image_name} 不存在,跳过文件 {filename}")
                continue

            image_info = {
                "id": image_id,
                "width": image_width,
                "height": image_height,
                "file_name": image_name
            }
            data['images'].append(image_info)

            # 读取文本文件
            with open(os.path.join(input_dir, filename), 'r') as file:
                lines = file.readlines()

            # 遍历文件中的每一行
            for line in lines:
                parts = line.strip().split()
                if len(parts) != 5:
                    print(f"警告: 文件 {filename} 中的行格式错误: {line.strip()}")
                    continue  # 跳过格式错误的行

                category_id = int(parts[0])
                x_center = float(parts[1]) * image_width
                y_center = float(parts[2]) * image_height
                bbox_width = float(parts[3]) * image_width
                bbox_height = float(parts[4]) * image_height

                x_min = int(x_center - bbox_width / 2)
                y_min = int(y_center - bbox_height / 2)
                bbox = [x_min, y_min, bbox_width, bbox_height]
                area = bbox_width * bbox_height

                annotation_info = {
                    "id": annotation_id,
                    "image_id": image_id,
                    "category_id": category_id,
                    "bbox": bbox,
                    "area": area,
                    "iscrowd": 0
                }
                data['annotations'].append(annotation_info)
                annotation_id += 1

            image_id += 1

    # 保存到 JSON 文件
    with open(output_json, 'w') as json_file:
        json.dump(data, json_file, indent=4)

    print(f"转换完成,输出文件保存为 {output_json}")


# 调用函数
input_dir = './personDataset/train'
output_json = './trainval.json'
txt_to_json(input_dir, output_json)

input_dir = './personDataset/test'
output_json = './test.json'
txt_to_json(input_dir, output_json)

结果:

![](https://i-blog.csdnimg.cn/direct/7e18c88f355340d5aaae4d466bda827e.png![在这里插入图片描述](https://i-blog.csdnimg.cn/direct/f7095e1cd23b4431bf795f934efcd0eb.png)

相关推荐
B站计算机毕业设计之家10 小时前
多模态项目:Python人脸表情系统 CNN算法 神经网络+Adaboost定位+PyQt5界面 源码+文档 深度学习实战✅
python·深度学习·神经网络·opencv·yolo·计算机视觉·情绪识别
dlraba80217 小时前
YOLO 目标检测算法全解析:原理、分类与性能指标
算法·yolo·目标检测
荼蘼19 小时前
【YOLO 模型入门】(1)一文读懂 YOLO:从核心概念到检测原理
yolo
小关会打代码1 天前
深度学习之YOLO系列了解基本知识
人工智能·深度学习·yolo
zy_destiny2 天前
【工业场景】用YOLOv8实现反光衣识别
人工智能·python·yolo·机器学习·计算机视觉
算法打盹中2 天前
计算机视觉:基于 YOLO 的轻量级目标检测与自定义目标跟踪原理与代码框架实现
图像处理·yolo·目标检测·计算机视觉·目标跟踪
小关会打代码2 天前
深度学习之YOLO系列YOLOv1
人工智能·深度学习·yolo
B站计算机毕业设计之家3 天前
智慧交通项目:Python+YOLOv8 实时交通标志系统 深度学习实战(TT100K+PySide6 源码+文档)✅
人工智能·python·深度学习·yolo·计算机视觉·智慧交通·交通标志
听风吹雨yu4 天前
RK3588从数据集到训练到部署YoloV8
linux·yolo·开源·rk3588·rknn
weixin_456904274 天前
YOLOv11安卓目标检测App完整开发指南
android·yolo·目标检测