Python實現xml格式轉txt格式的範例程式碼

import os
import xml.etree.ElementTree as ET
import io
find_path = './train/'    #xml所在的檔案
savepath='./labels/'   #儲存檔案

class Voc_Yolo(object):
    def __init__(self, find_path):
        self.find_path = find_path
    def Make_txt(self, outfile):
        out = open(outfile,'w') 
        print("建立成功：{}".format(outfile))
        return out
    def Work(self, count):
    #找到檔案路徑
        for root, dirs, files in os.walk(self.find_path):
        #找到檔案目錄中每一個xml檔案
            for file in files:
            #記錄處理過的檔案
                count += 1
                #輸入、輸出檔案定義
                input_file = find_path + file
                outfile = savepath+file[:-4]+'.txt'
                #新建txt檔案，確保檔案正常儲存
                out = self.Make_txt(outfile)
                #分析xml樹，取出w_image、h_image
                tree=ET.parse(input_file)
                root=tree.getroot()
                size=root.find('size')
                w_image=float(size.find('width').text)
                h_image=float(size.find('height').text)
                #繼續提取有效資訊來計算txt中的四個資料
                for obj in root.iter('object'):
                #將型別提取出來，不同目標型別不同，本文僅有一個類別->0
                    classname=obj.find('name').text
                    cls_id = classname
                    xmlbox=obj.find('bndbox')
                    x_min=float(xmlbox.find('xmin').text)
                    x_max=float(xmlbox.find('xmax').text)
                    y_min=float(xmlbox.find('ymin').text)
                    y_max=float(xmlbox.find('ymax').text)
                    #計算公式
                    x_center=((x_min+x_max)/2-1)/w_image
                    y_center=((y_min+y_max)/2-1)/h_image
                    w=(x_max-x_min)/w_image
                    h=(y_max-y_min)/h_image
                    #檔案寫入
                    out.write(str(cls_id)+" "+str(x_center)+" "+str(y_center)+" "+str(w)+" "+str(h)+'n')
                out.close()
        return count
if __name__ == "__main__":
    data = Voc_Yolo(find_path)
    number = data.Work(0)
    print(number)