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

这篇博客介绍如何使用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
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值