计算机视觉：经典数据格式(VOC、YOLO、COCO)解析与转换(附代码)

news/2025/2/24 17:34:52

第一章：计算机视觉中图像的基础认知
第二章：计算机视觉：卷积神经网络(CNN)基本概念(一)
第三章：计算机视觉：卷积神经网络(CNN)基本概念(二)
第四章：搭建一个经典的LeNet5神经网络(附代码)
第五章：计算机视觉：神经网络实战之手势识别(附代码)
第六章：计算机视觉：目标检测从简单到容易(附代码)
第七章：MTCNN 人脸检测技术揭秘：原理、实现与实战(附代码)
第八章：探索YOLO技术：目标检测的高效解决方案
第九章：计算机视觉：主流数据集整理
第十章：生成对抗网络(GAN)：从概念到代码实践(附代码)
第十一章：计算机视觉：经典数据格式(VOC、YOLO、COCO)解析与转换(附代码)
第十二章：计算机视觉：YOLOv11遥感图像目标检测(附代码)

在计算机视觉（CV）领域，无论是进行目标检测、图像分类还是其他任务，理解如何处理不同格式的数据集以及掌握训练过程中涉及的关键指标至关重要。本文将探讨三种经典的数据格式（VOC、YOLO、COCO）

一、VOC 格式

VOC（Visual Object Classes）格式是一种广泛应用于目标检测任务的数据标注标准，尤其常见于PASCAL VOC挑战赛中。它使用XML文件来存储图像中的对象位置信息和类别信息。

文件结构与内容

每个图像对应一个XML文件，该文件包含了图像的基本信息以及图像中每个对象的位置和类别标签。以下是一个典型的VOC格式XML文件的内容示例：

<annotation>
    <folder>images</folder>
    <filename>000001.jpg</filename>
    <size>
        <width>500</width>
        <height>375</height>
        <depth>3</depth>
    </size>
    <object>
        <name>dog</name>
        <pose>Left</pose>
        <truncated>1</truncated>
        <difficult>0</difficult>
        <bndbox>
            <xmin>263</xmin>
            <ymin>211</ymin>
            <xmax>324</xmax>
            <ymax>339</ymax>
        </bndbox>
    </object>
    <object>
        <name>person</name>
        <pose>Unspecified</pose>
        <truncated>0</truncated>
        <difficult>0</difficult>
        <bndbox>
            <xmin>159</xmin>
            <ymin>59</ymin>
            <xmax>281</xmax>
            <ymax>287</ymax>
        </bndbox>
    </object>
</annotation>

关键元素说明

  • <folder>：包含图像的文件夹名称。
  • <filename>：图像文件名。
  • <size>：描述图像尺寸，包括宽度、高度和深度（通常是3表示RGB图像）。
  • <object>：每个对象的信息块，可以有多个，每个对象包含：
    • <name>：对象类别名称。
    • <pose>：拍摄时物体的姿态。
    • <truncated>：指示物体是否被裁剪（部分位于图像外）。
    • <difficult>：指示物体是否难以识别。
    • <bndbox>：边界框坐标，包括：
      • <xmin>, <ymin>：边界框左上角的绝对坐标（像素值）。
      • <xmax>, <ymax>：边界框右下角的绝对坐标（像素值）。

处理VOC数据的Python代码示例

下面是一个简单的例子，展示如何读取并解析VOC格式的XML文件，并提取其中的对象信息：

from xml.etree import ElementTree

def parse_voc_xml(file_path):
    tree = ElementTree.parse(file_path)
    root = tree.getroot()
    
    # 获取图像尺寸
    img_width = int(root.find("size/width").text)
    img_height = int(root.find("size/height").text)
    
    objects = []
    for obj in root.findall("object"):
        name = obj.find("name").text
        xmin = int(obj.find("bndbox/xmin").text)
        ymin = int(obj.find("bndbox/ymin").text)
        xmax = int(obj.find("bndbox/xmax").text)
        ymax = int(obj.find("bndbox/ymax").text)
        
        objects.append({
            "name": name,
            "bbox": [xmin, ymin, xmax, ymax]
        })
    
    return img_width, img_height, objects

# 使用示例
file_path = "path/to/voc_annotation.xml"
width, height, objs = parse_voc_xml(file_path)
print(f"Image width: {width}, height: {height}")
for obj in objs:
    print(obj)

此代码段展示如何从给定的VOC格式XML文件中提取图像尺寸和每个对象的位置及类别信息。

YOLO__112">二、YOLO 格式

YOLO（You Only Look Once）是一种流行的目标检测算法，它使用一种特定的数据标注格式来描述图像中的对象位置和类别信息。与VOC或COCO等其他数据格式不同，YOLO格式采用文本文件（.txt）存储每个图像的标注信息，这些信息包括对象的类别ID及其边界框的位置坐标。

文件结构与内容

对于每张图像，YOLO格式会有一个对应的文本文件，该文件中每一行代表一个对象，并且包含五个数值：

  1. 类别ID（cls_id）
  2. 边界框中心点的x坐标（x_center）
  3. 边界框中心点的y坐标（y_center）
  4. 边界框的宽度（w）
  5. 边界框的高度（h）

所有坐标都是相对坐标，即相对于图像宽度和高度的比例值（0到1之间的小数），而不是绝对像素值。以下是YOLO格式的一个简单示例：

假设有一张分辨率为640x480的图片，其中包含两个对象：一只狗和一个人。相应的YOLO格式标注文件可能如下所示：

0 0.500000 0.600000 0.250000 0.300000 # 狗
1 0.300000 0.200000 0.100000 0.150000 # 人
  • 第一行表示“狗”的类别ID为0，其边界框中心位于图像宽度的50%、高度的60%，宽度占整个图像宽度的25%，高度占30%。
  • 第二行表示“人”的类别ID为1，其边界框中心位于图像宽度的30%、高度的20%，宽度占整个图像宽度的10%，高度占15%。
    在这里插入图片描述

处理YOLO数据的Python代码示例

以下是一个简单的例子，展示如何将VOC格式转换为YOLO格式，并读取YOLO格式的数据。

from xml.etree import ElementTree

def voc_to_yolo(voc_file_path, output_file_path, label2idx):
    tree = ElementTree.parse(voc_file_path)
    root = tree.getroot()
    
    img_width = int(root.find("size/width").text)
    img_height = int(root.find("size/height").text)
    
    with open(output_file_path, 'w') as f:
        for obj in root.findall("object"):
            name = obj.find("name").text
            cls_id = label2idx[name]
            
            xmin = int(obj.find("bndbox/xmin").text)
            ymin = int(obj.find("bndbox/ymin").text)
            xmax = int(obj.find("bndbox/xmax").text)
            ymax = int(obj.find("bndbox/ymax").text)
            # 这是计算边界框左上角和右下角的x坐标的平均值，即边界框中心点的x坐标（以像素为单位）。
            x_center = (xmin + xmax) / 2.0 / img_width
            # 这是计算边界框左上角和右下角的y坐标的平均值，即边界框中心点的y坐标（以像素为单位）。
            y_center = (ymin + ymax) / 2.0 / img_height
            width = (xmax - xmin) / float(img_width)
            height = (ymax - ymin) / float(img_height)
            
            line = f"{cls_id} {x_center:.6f} {y_center:.6f} {width:.6f} {height:.6f}\n"
            f.write(line)

# 示例用法
label2idx = {"dog": 0, "person": 1}
voc_file_path = "path/to/voc_annotation.xml"
output_file_path = "path/to/output.txt"
voc_to_yolo(voc_file_path, output_file_path, label2idx)

在YOLO格式中，边界框的坐标是以相对坐标的形式表示的，而不是绝对像素值。具体来说，x_center 和 y_center 分别代表边界框中心点相对于图像宽度和高度的比例值（范围从0到1），而 w 和 h 分别代表边界框的宽度和高度相对于图像宽度和高度的比例值。

公式解释

x_center = (xmin + xmax) / 2.0 / img_width
y_center = (ymin + ymax) / 2.0 / img_height

计算边界框中心点的相对坐标

  1. 计算边界框中心点的绝对坐标：

    • (xmin + xmax) / 2.0：这是计算边界框左上角和右下角的x坐标的平均值，即边界框中心点的x坐标（以像素为单位）。
    • (ymin + ymax) / 2.0：这是计算边界框左上角和右下角的y坐标的平均值，即边界框中心点的y坐标（以像素为单位）。

  2. 转换为相对坐标：

    • / img_width：将边界框中心点的x坐标除以图像的宽度，得到一个比例值（范围从0到1）。例如，如果边界框中心点的x坐标是320像素，而图像的宽度是640像素，则 x_center 的值为 320 / 640 = 0.5。
    • / img_height：将边界框中心点的y坐标除以图像的高度，得到一个比例值（范围从0到1）。例如，如果边界框中心点的y坐标是240像素，而图像的高度是480像素，则 y_center 的值为 240 / 480 = 0.5。

示例

假设有一张分辨率为640x480的图片，其中有一个对象的边界框坐标如下：

  • xmin = 100
  • ymin = 150
  • xmax = 300
  • ymax = 350

根据上述公式计算：

  1. 计算边界框中心点的绝对坐标：

    • x_center_abs = (100 + 300) / 2.0 = 200
    • y_center_abs = (150 + 350) / 2.0 = 250

  2. 转换为相对坐标：

    • x_center_rel = 200 / 640 ≈ 0.3125
    • y_center_rel = 250 / 480 ≈ 0.5208

因此，在YOLO格式的标注文件中，该对象的标注信息可能如下所示：

0 0.3125 0.5208 0.3125 0.4167

其中：

  • 0 是类别ID。
  • 0.3125 是边界框中心点的x坐标相对于图像宽度的比例值。
  • 0.5208 是边界框中心点的y坐标相对于图像高度的比例值。
  • 0.3125 是边界框宽度相对于图像宽度的比例值（(300 - 100) / 640 = 200 / 640 ≈ 0.3125）。
  • 0.4167 是边界框高度相对于图像高度的比例值（(350 - 150) / 480 = 200 / 480 ≈ 0.4167）。

读取YOLO格式数据

def read_yolo_annotations(file_path):
    annotations = []
    with open(file_path, 'r') as f:
        lines = f.readlines()
        for line in lines:
            parts = line.strip().split()
            cls_id = int(parts[0])
            x_center, y_center, w, h = map(float, parts[1:])
            annotations.append({
                "cls_id": cls_id,
                "bbox": [x_center, y_center, w, h]
            })
    return annotations

# 示例用法
file_path = "path/to/yolo_annotation.txt"
annotations = read_yolo_annotations(file_path)
for annotation in annotations:
    print(annotation)

通过上述示例，可以轻松地在VOC格式和YOLO格式之间进行转换，并读取YOLO格式的数据。这对于准备训练数据集或进行数据分析非常有用。

COCO__255">三、COCO 格式

COCO（Common Objects in Context）格式是一种广泛用于计算机视觉任务，特别是目标检测、分割和关键点检测的数据标注标准。它采用JSON文件来存储图像及其对应的注释信息，具有高度结构化的特点，支持复杂的多对象标注。

文件结构与内容

COCO格式的JSON文件通常包含以下几个主要部分：

  1. images: 包含图像的基本信息。
  2. annotations: 描述图像中的每个对象或区域的信息。
  3. categories: 定义所有可能的对象类别。

以下是一个简化的COCO格式JSON文件示例：

{
    "images": [
        {
            "id": 0,
            "width": 640,
            "height": 480,
            "file_name": "000000000009.jpg"
        }
    ],
    "annotations": [
        {
            "id": 1,
            "image_id": 0,
            "category_id": 1,
            "bbox": [100, 150, 200, 200],
            "area": 40000,
            "iscrowd": 0
        },
        {
            "id": 2,
            "image_id": 0,
            "category_id": 2,
            "bbox": [300, 200, 100, 150],
            "area": 15000,
            "iscrowd": 0
        }
    ],
    "categories": [
        {
            "id": 1,
            "name": "person",
            "supercategory": "person"
        },
        {
            "id": 2,
            "name": "dog",
            "supercategory": "animal"
        }
    ]
}
  • images: 每个元素包含一个图像的信息，如ID、宽度、高度和文件名。
  • annotations: 每个元素描述一个对象的位置（通过边界框bbox）、面积area、是否为群体对象iscrowd等信息。
  • categories: 定义了所有可能的对象类别及其ID。

关键字段解释

  • bbox: 边界框的坐标，格式为 [x, y, width, height]，其中x和y是边界框左上角的绝对坐标（像素值），width和height是边界框的宽度和高度（同样以像素为单位）。
  • area: 对象的面积，对于目标检测任务，这通常是边界框的面积（宽度乘以高度）。
  • iscrowd: 标记该对象是否为群体对象（例如一群人聚集在一起）。如果为1，则表示该对象是一个群体；如果为0，则表示单独的对象。

处理COCO数据的Python代码示例

下面是一个简单的例子，展示如何读取并解析COCO格式的JSON文件，并提取其中的对象信息：

import json

def parse_coco_json(file_path):
    with open(file_path, 'r') as f:
        data = json.load(f)
    
    images = {img['id']: img for img in data['images']}
    categories = {cat['id']: cat for cat in data['categories']}
    
    annotations = []
    for ann in data['annotations']:
        image_info = images[ann['image_id']]
        category_info = categories[ann['category_id']]
        
        annotation = {
            "image_id": ann['image_id'],
            "filename": image_info['file_name'],
            "category_id": ann['category_id'],
            "category_name": category_info['name'],
            "bbox": ann['bbox'],
            "area": ann['area']
        }
        annotations.append(annotation)
    
    return annotations

# 示例用法
file_path = "path/to/coco_annotation.json"
annotations = parse_coco_json(file_path)
for annotation in annotations:
    print(annotation)

输出结果：

{'image_id': 0, 'filename': '000000000009.jpg', 'category_id': 1, 'category_name': 'person', 'bbox': [100, 150, 200, 200], 'area': 40000}
{'image_id': 0, 'filename': '000000000009.jpg', 'category_id': 2, 'category_name': 'dog', 'bbox': [300, 200, 100, 150], 'area': 15000}

上述代码首先将images和categories转换为字典以便快速查找，然后遍历所有的annotations，从中提取相关信息并打印出来。


http://www.niftyadmin.cn/n/5864652.html

相关文章

MySQL 单表访问方法详解

MySQL 单表访问方法详解

单表访问 MySQL 单表访问方法详解&#xff1a;高效查询之道**一、 查询执行基础****二、 访问方法 (Access Method) 概念****三、 具体访问方法 (从最优到最差)****四、 注意事项****五、 总结与优化建议****六、 电商网站数据存储应用示例****七、 数据备份与恢复模型 (补充)*…
阅读更多...
Python安全之反序列化——pickle/cPickle

Python安全之反序列化——pickle/cPickle

一&#xff0e; 概述 Python中有两个模块可以实现对象的序列化&#xff0c;pickle和cPickle&#xff0c;区别在于cPickle是用C语言实现的&#xff0c;pickle是用纯python语言实现的&#xff0c;用法类似&#xff0c;cPickle的读写效率高一些。使用时一般先尝试导入cPickle&…
阅读更多...
基于AT89C52单片机的出租车计价器

基于AT89C52单片机的出租车计价器

点击链接获取Keil源码与Project Backups仿真图&#xff1a; https://download.csdn.net/download/qq_64505944/90419909?spm1001.2014.3001.5501 C17 部分参考设计如下&#xff1a; 摘要 随着城市交通行业的迅速发展&#xff0c;出租车作为最主要的城市公共交通工具之一…
阅读更多...
http 协议在互联网中扮演着怎样的角色？

http 协议在互联网中扮演着怎样的角色？

互联网各领域资料分享专区(不定期更新)： Sheet 正文 HTTP（超文本传输协议）在互联网中扮演着核心通信协议的角色，是万维网（World Wide Web）的基础技术之一。 1. 客户端-服务器交互的桥梁 浏览器与服务器的通信语言：HTTP定义了浏览器（客户端）如何向服务器请求资源（如…
阅读更多...
Python 基本语法的详细解释

Python 基本语法的详细解释

目录 &#xff08;1&#xff09;注释 &#xff08;2&#xff09;缩进 &#xff08;3&#xff09;变量和数据类型 变量定义 数据类型 &#xff08;4&#xff09;输入和输出 输出&#xff1a;print() 函数 输入&#xff1a;input() 函数 &#xff08;1&#xff09;注释 注…
阅读更多...
黑马点评 面试话术

黑马点评 面试话术

MybatisPlus session技术会把jsessionid自动写到cookie里 ThreadLocal保证线程安全 用springmvc的自定义拦截器把登出用户 查看详情 获取当前用户并且返回 上传操作 登录查看详情 以下是不拦截的路径 在tomcat负载均衡时 如果不使用redis 直接用相互拷贝 1 浪费空间 2 如果此时…
阅读更多...
软件需求管理办法，软件开发管理指南（Word原件）

软件需求管理办法，软件开发管理指南（Word原件）

1. 目的 2. 适用范围 3. 参考文件 4. 术语和缩写 5. 需求获取的方式 5.1. 与用户交谈向用户提问题 5.1.1. 访谈重点注意事项 5.1.2. 访谈指南 5.2. 参观用户的工作流程 5.3. 向用户群体发调查问卷 5.4. 已有软件系统调研 5.5. 资料收集 5.6. 原型系统调研 5.6.1. …
阅读更多...
JAVA-Exploit编写(13-15)--JAVAFX-GUI检测工具编写实现

JAVA-Exploit编写(13-15)--JAVAFX-GUI检测工具编写实现

目录 一,JAVAFX-GUI单个漏洞检测编写 1.1 绑定事件 1.2 Thinkphp5_Rce编写 1.3 编写利用类 1.4 Thinkphp2x_Rce编写 1.5 单个漏洞检测GUI工具完整代码 二,JAVAFX-GUI单个漏洞批量检测编写 2.1 编写利用反射类 2.2 批量检测漏洞完整GUI工具代码 三,JAVAFX-GUI…
阅读更多...
最新文章

Copyright @ 2022~2023