数据集 VOC转YOLO5,xml to txt

做yolo5目标检测训练时,很多小伙伴遇到了xml转txt数据集的问题,我把转换代码贴到下面,有需要自取。


```python
import xml.etree.ElementTree as ET
import os
from os import listdir, getcwd
from os.path import join


outpath = "E:\\Documents\\person\\labels"
xmlpaths = "E:\\Documents\\person\\xml"

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


def xmlTotxt():
    xmls = os.listdir(xmlpaths)
    for xml in xmls:
        xmlpath = os.path.join(xmlpaths, xml)
        tree = ET.parse(xmlpath)
        root = tree.getroot()
        size = root.find('size')
        w = int(size.find('width').text)
        h = int(size.find('height').text)
        out_file = os.path.join(outpath, xml.split('.')[0] + '.txt')
        out_txt_f = open(out_file, 'w')
        for obj in root.iter('object'):
            cls_id = 1
            xmlbox = obj.find('bndbox')
            b = (float(xmlbox.find('xmin').text), float(xmlbox.find('xmax').text), float(xmlbox.find('ymin').text),
                    float(xmlbox.find('ymax').text))
            bb = convert((w, h), b)
            out_txt_f.write(str(cls_id) + " " + " ".join([str(a) for a in bb]) + '\n')
        out_txt_f.close()
    
def main():
    xmlTotxt()
    print("finished")

if __name__ == "__main__":
    main()



评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

清风. 

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值