COCO/YOLO/VOC 3 大格式互转：1 个脚本实现 6 种转换路径-尧图网络科技

COCO/YOLO/VOC 三大格式互转：一站式脚本实现全路径转换

1. 目标检测数据格式的现状与挑战

在计算机视觉领域，目标检测任务的成功很大程度上依赖于高质量的数据集。COCO、YOLO和VOC作为当前主流的三种数据格式，各自拥有独特的结构和应用场景。COCO格式以其丰富的标注信息和标准化评估指标著称，YOLO格式则因其简洁高效而广受欢迎，而VOC作为经典格式仍然被许多传统项目所采用。

实际工作中，研究人员和工程师经常面临以下痛点：

多模型验证需求：不同框架对输入格式要求不同（如YOLOv5需要YOLO格式，MMDetection偏好COCO格式）
数据集复用困难：公开数据集往往只提供单一格式（如COCO），需要转换为其他格式才能使用
标注工具差异：不同标注工具输出的格式各异（LabelImg生成VOC，CVAT支持COCO等）
迁移学习障碍：预训练模型使用不同格式，导致微调时需要进行格式转换

传统解决方案通常需要编写多个独立脚本分别处理不同方向的转换，不仅效率低下，而且容易出错。本文将介绍一个统一的Python脚本框架，实现三大格式间的六种可能转换路径。

2. 三大格式深度解析与技术对比

2.1 COCO格式剖析

COCO（Common Objects in Context）格式采用JSON结构存储标注信息，主要包含以下核心字段：

{ "images": [ { "id": int, "width": int, "height": int, "file_name": str, "license": int, "coco_url": str } ], "annotations": [ { "id": int, "image_id": int, "category_id": int, "bbox": [x,y,width,height], "area": float, "iscrowd": 0 or 1 } ], "categories": [ { "id": int, "name": str, "supercategory": str } ] }

关键特点：

使用相对路径引用图像文件
边界框采用[x,y,width,height]格式（绝对像素值）
支持多任务标注（检测、分割、关键点等）
类别ID可以不连续

2.2 YOLO格式详解

YOLO格式的核心是每个图像对应一个.txt文件，每行表示一个对象：

<object-class> <x_center> <y_center> <width> <height>

转换公式（像素坐标→归一化坐标）：

def convert(size, box): dw = 1./size[0] dh = 1./size[1] x = (box[0] + box[2]/2.0) * dw y = (box[1] + box[3]/2.0) * dh w = box[2] * dw h = box[3] * dh return (x,y,w,h)

文件结构示例：

dataset/ ├── images/ │ ├── train/ │ └── val/ └── labels/ ├── train/ └── val/

2.3 VOC格式解析

VOC格式采用XML文件存储标注，典型结构如下：

<annotation> <filename>image.jpg</filename> <size> <width>640</width> <height>480</height> <depth>3</depth> </size> <object> <name>person</name> <bndbox> <xmin>100</xmin> <ymin>200</ymin> <xmax>300</xmax> <ymax>400</ymax> </bndbox> </object> </annotation>

2.4 格式对比表格

特性	COCO	YOLO	VOC
存储格式	JSON	TXT	XML
坐标系统	绝对像素[x,y,w,h]	归一化[x_center,y_center,w,h]	绝对像素[xmin,ymin,xmax,ymax]
文件结构	单文件集中存储	图像与标注文件分离	每个图像对应XML文件
标注复杂度	高（支持多任务）	低（仅检测）	中（检测+分类）
扩展性	优秀	一般	良好
主流框架支持	MMDetection, Detectron2	YOLO系列	传统框架

3. 统一转换框架设计与实现

3.1 核心架构设计

我们的转换框架采用模块化设计，主要包含以下组件：

converter/ ├── __init__.py ├── base.py # 抽象基类 ├── coco.py # COCO相关转换 ├── yolo.py # YOLO相关转换 ├── voc.py # VOC相关转换 └── utils.py # 通用工具函数

转换流程：

解析源格式文件
转换为中间通用表示
生成目标格式文件
验证转换结果

3.2 核心转换代码实现

以下是关键转换函数的代码示例：

class COCOToYOLOConverter: def __init__(self, json_path, output_dir): self.json_path = json_path self.output_dir = output_dir os.makedirs(output_dir, exist_ok=True) def convert_bbox(self, size, box): """将COCO bbox转换为YOLO格式""" dw = 1. / size[0] dh = 1. / size[1] x = box[0] + box[2] / 2.0 y = box[1] + box[3] / 2.0 w = box[2] h = box[3] x = x * dw w = w * dw y = y * dh h = h * dh return (x, y, w, h) def convert(self): with open(self.json_path) as f: data = json.load(f) # 创建类别映射 categories = {cat['id']: idx for idx, cat in enumerate(data['categories'])} # 处理每个图像 for img in data['images']: img_id = img['id'] file_name = os.path.splitext(img['file_name'])[0] txt_path = os.path.join(self.output_dir, f"{file_name}.txt") annotations = [ann for ann in data['annotations'] if ann['image_id'] == img_id] with open(txt_path, 'w') as f_txt: for ann in annotations: class_id = categories[ann['category_id']] box = self.convert_bbox((img['width'], img['height']), ann['bbox']) line = f"{class_id} {box[0]:.6f} {box[1]:.6f} {box[2]:.6f} {box[3]:.6f}\n" f_txt.write(line)

3.3 六种转换路径实现

框架支持的完整转换矩阵：

源格式 → 目标格式	COCO	YOLO	VOC
COCO	-	`coco2yolo`	`coco2voc`
YOLO	`yolo2coco`	-	`yolo2voc`
VOC	`voc2coco`	`voc2yolo`	-

特殊处理场景：

处理VOC的difficult标记
处理COCO的iscrowd标注
YOLO类别ID的连续性保证
图像路径的自动修正

4. 高级功能与最佳实践

4.1 批处理与增量转换

框架支持以下高级功能：

# 批量转换示例 converter = BatchConverter( input_dir="datasets/coco", output_dir="datasets/yolo", input_format="coco", output_format="yolo" ) converter.process() # 处理整个目录 # 增量转换示例 converter = IncrementalConverter( input_dir="datasets/voc", output_dir="datasets/coco", input_format="voc", output_format="coco" ) converter.watch() # 监控目录变化并自动转换新文件

4.2 验证与可视化

为确保转换质量，框架提供验证工具：

validator = ConversionValidator( original_path="datasets/coco/annotations.json", converted_path="datasets/yolo/labels", original_format="coco", converted_format="yolo" ) report = validator.validate() # 可视化差异 visualizer = AnnotationVisualizer( image_dir="datasets/coco/images", original_ann="datasets/coco/annotations.json", converted_ann="datasets/yolo/labels", output_dir="comparison" ) visualizer.generate_comparison()

4.3 性能优化技巧

处理大型数据集时：

使用内存映射处理大JSON文件
多进程并行转换
增量写入避免内存爆炸

# 使用ijson流式处理大COCO文件 import ijson def stream_coco(json_path): with open(json_path, 'rb') as f: images = ijson.items(f, 'images.item') for img in images: yield img

5. 实战应用与问题排查

5.1 典型应用场景

场景一：使用COCO预训练模型微调YOLO格式数据

将YOLO格式转换为COCO格式
使用MMDetection框架微调模型
将预测结果转换回YOLO格式

场景二：多框架模型集成

graph LR A[VOC数据集] -->|voc2yolo| B(YOLOv5训练) A -->|voc2coco| C(Detectron2评估) B -->|yolo2coco| D[统一评估指标] C --> D

5.2 常见问题解决方案

问题1：类别ID不一致

解决方案：使用--class-map参数指定自定义映射文件

问题2：图像路径错误

解决方案：添加--image-root参数指定图像根目录

问题3：坐标越界

添加边界检查代码：

def safe_convert(size, box): x, y, w, h = box # 确保坐标在有效范围内 x = max(0, min(x, size[0]-1)) y = max(0, min(y, size[1]-1)) w = max(1, min(w, size[0]-x)) h = max(1, min(h, size[1]-y)) return (x, y, w, h)