zoukankan      html  css  js  c++  java
  • VOC标签转化为YOLO标签

    参考darknet自带的voc_label.py

    import xml.etree.ElementTree as ET
    import pickle
    import os
    from os import listdir, getcwd
    from os.path import join
    
    classes = ["dog", "cat"]
    
    
    def convert(size, box):
        dw = 1./(size[0])
        dh = 1./(size[1])
        x = (box[0] + box[1])/2.0 - 1
        y = (box[2] + box[3])/2.0 - 1
        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 convert_annotation(rootpath,xmlname):
        xmlpath = rootpath + '/labels'
        xmlfile = os.path.join(xmlpath,xmlname)
        with open(xmlfile, "r") as in_file:
          txtname = xmlname[:-4]+'.txt'
          txtpath = rootpath + '/labelYOLOs'
          if not os.path.exists(txtpath):
            os.makedirs(txtpath)
          txtfile = os.path.join(txtpath,txtname)
          with open(txtfile, "w+") as out_file:
            tree=ET.parse(in_file)
            root = tree.getroot()
            size = root.find('size')
            w = int(size.find('width').text)
            h = int(size.find('height').text)
            out_file.truncate()
            for obj in root.iter('object'):
                difficult = obj.find('difficult').text
                cls = obj.find('name').text
                if cls not in classes or int(difficult)==1:
                    continue
                cls_id = classes.index(cls)
                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_file.write(str(cls_id) + " " + " ".join([str(a) for a in bb]) + '
    ')
    
    
    if __name__ == "__main__":
        rootpath='/home/aimhabo/YOLOv3SMP/data'
        xmlpath=rootpath+'/labels'
        list=os.listdir(xmlpath)
        for i in range(0,len(list)) :
            path = os.path.join(xmlpath,list[i])
            if ('.xml' in path)or('.XML' in path):
                convert_annotation(rootpath,list[i])
                print('done', i)
            else:
                print('not xml file',i)
  • 相关阅读:
    POJ 3253 Fence Repair
    POJ 1328 Radar Installation
    bzoj 4010: [HNOI2015]菜肴制作
    bzoj 4008: [HNOI2015]亚瑟王
    UVA 1451 Average
    UVA 1481 Genome Evolution
    HDU 1542 Atlantis
    UVA 11419 SAM I AM
    UVA 11762 Race to 1
    P2209 [USACO13OPEN]燃油经济性Fuel Economy
  • 原文地址:https://www.cnblogs.com/aimhabo/p/8949837.html
Copyright © 2011-2022 走看看