python 将yaml标签文件转化为xml格式的标签文件

mark_leiliu

于 2018-05-23 21:31:40 发布

阅读量3.1k

点赞数 2

CC 4.0 BY-SA版权

分类专栏： python 文章标签： yaml to xml

本文链接：https://blog.youkuaiyun.com/T1243_3/article/details/80426589

这篇博客介绍如何使用Python将yaml标签文件转换为xml格式，以适应SSD和Faster R-CNN等深度学习代码的数据集需求。作者提供了label_images.py和WriteFileXml.py两个程序，详细说明了转换过程，并展示了转换后的xml文件格式。

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

最近在跑SSD和Faster R-CNN深度学习代码，下载了一些数据集，但是这些数据集标签文件不是xml格式文件，而是yaml文件，虽然网上有在线转化的工具，但是这种做法对我来说显然是很低效率的。为了提高效率，自己写了相关的代码。现在分享给大家。

感谢Bosch Small Traffic Lights Dataset 提供的部分代码。

以下是label_images.py程序：

#!/usr/bin/env python
 """
Example usage:
    python label_images.py input.yaml [output_folder]
"""
import sys
import os
import cv2
from read_label_file import get_all_labels
from WriteFileXml import writeInfoToXml

def ir(some_value):
    """Int-round function for short array indexing """
    return int(round(some_value))

def show_label_images(input_yaml, output_folder=None):
    images = get_all_labels(input_yaml)

    if output_folder is not None:
        if not os.path.exists(output_folder):
            os.makedirs(output_folder)

    for i, image_dict in enumerate(images):
        #print image_dict['boxes']
        image = cv2.imread(image_dict['path'])
        img_size=image.shape

        if image is None:
            rai