深度学习之数据集转换脚本(voc的xml格式转yolo的txt格式篇)

在做目标检测时,有些代码库采用的是yolo的txt格式,有些则采用的是voc的xml格式,下面提供了一个脚本,可以自动将voc格式的数据集转化为yolo格式的数据集。

voc格式

voc格式的数据集为以下格式,其中Main文件夹里面的txt文件是按比例划分好的训练、验证和测试集,文件里面的每一行代表一张图片(若没有提前划分,可以使用voc数据集划分脚本来划分):

VOC2007
   |__Annotations
   |__JPEGImages
   |__ImageSets
          |__Main
              |__train
              |__val
              |__test
              |__trainval

yolo格式

yolo格式下一般分为images和labels文件夹,在这两个文件夹里面又包含有train,val(以及test)文件夹,如下面所示:

转换脚本 

1. xml转txt

接下来采用脚本,自动将voc格式转化为yolo格式,设置好xml文件的路径和要保存的txt文件路径,执行后便可以转换。(转换后原xml文件依旧存在)

import os
import glob
import xml.etree.ElementTree a
  • 3
    点赞
  • 28
    收藏
    觉得还不错? 一键收藏
  • 5
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值