from xml.dom.minidom import Document
import os
import cv2
# def makexml(txtPath, xmlPath, picPath):  # txt所在文件夾路徑，xml文件保存路徑，圖片所在文件夾路徑
def makexml(picPath, txtPath, xmlPath):  # txt所在文件夾路徑，xml文件保存路徑，圖片所在文件夾路徑
    """此函數(shù)用于將yolo格式txt標(biāo)注文件轉(zhuǎn)換為voc格式xml標(biāo)注文件
    """
    dic = {'0': "pedestrian",  # 創(chuàng)建字典用來對(duì)類型進(jìn)行轉(zhuǎn)換
           '1': "people",  # 此處的字典要與自己的classes.txt文件中的類對(duì)應(yīng)，且順序要一致
           '2': "bicycle",
           '3': "car",
           '4': "van",
           '5': "truck",
           '6': "tricycle",
           '7': "awning-tricycle",
           '8': "bus",
           '9': "motor",
           }
    files = os.listdir(txtPath)
    for i, name in enumerate(files):
        xmlBuilder = Document()
        annotation = xmlBuilder.createElement("annotation")  # 創(chuàng)建annotation標(biāo)簽
        xmlBuilder.appendChild(annotation)
        txtFile = open(txtPath + name)
        txtList = txtFile.readlines()
        img = cv2.imread(picPath + name[0:-4] + ".jpg")
        Pheight, Pwidth, Pdepth = img.shape
        folder = xmlBuilder.createElement("folder")  # folder標(biāo)簽
        foldercontent = xmlBuilder.createTextNode("driving_annotation_dataset")
        folder.appendChild(foldercontent)
        annotation.appendChild(folder)  # folder標(biāo)簽結(jié)束
        filename = xmlBuilder.createElement("filename")  # filename標(biāo)簽
        filenamecontent = xmlBuilder.createTextNode(name[0:-4] + ".jpg")
        filename.appendChild(filenamecontent)
        annotation.appendChild(filename)  # filename標(biāo)簽結(jié)束
        size = xmlBuilder.createElement("size")  # size標(biāo)簽
        width = xmlBuilder.createElement("width")  # size子標(biāo)簽width
        widthcontent = xmlBuilder.createTextNode(str(Pwidth))
        width.appendChild(widthcontent)
        size.appendChild(width)  # size子標(biāo)簽width結(jié)束
        height = xmlBuilder.createElement("height")  # size子標(biāo)簽height
        heightcontent = xmlBuilder.createTextNode(str(Pheight))
        height.appendChild(heightcontent)
        size.appendChild(height)  # size子標(biāo)簽height結(jié)束
        depth = xmlBuilder.createElement("depth")  # size子標(biāo)簽depth
        depthcontent = xmlBuilder.createTextNode(str(Pdepth))
        depth.appendChild(depthcontent)
        size.appendChild(depth)  # size子標(biāo)簽depth結(jié)束
        annotation.appendChild(size)  # size標(biāo)簽結(jié)束
        for j in txtList:
            oneline = j.strip().split(" ")
            object = xmlBuilder.createElement("object")  # object 標(biāo)簽
            picname = xmlBuilder.createElement("name")  # name標(biāo)簽
            namecontent = xmlBuilder.createTextNode(dic[oneline[0]])
            picname.appendChild(namecontent)
            object.appendChild(picname)  # name標(biāo)簽結(jié)束
            pose = xmlBuilder.createElement("pose")  # pose標(biāo)簽
            posecontent = xmlBuilder.createTextNode("Unspecified")
            pose.appendChild(posecontent)
            object.appendChild(pose)  # pose標(biāo)簽結(jié)束
            truncated = xmlBuilder.createElement("truncated")  # truncated標(biāo)簽
            truncatedContent = xmlBuilder.createTextNode("0")
            truncated.appendChild(truncatedContent)
            object.appendChild(truncated)  # truncated標(biāo)簽結(jié)束
            difficult = xmlBuilder.createElement("difficult")  # difficult標(biāo)簽
            difficultcontent = xmlBuilder.createTextNode("0")
            difficult.appendChild(difficultcontent)
            object.appendChild(difficult)  # difficult標(biāo)簽結(jié)束
            bndbox = xmlBuilder.createElement("bndbox")  # bndbox標(biāo)簽
            xmin = xmlBuilder.createElement("xmin")  # xmin標(biāo)簽
            mathData = int(((float(oneline[1])) * Pwidth + 1) - (float(oneline[3])) * 0.5 * Pwidth)
            xminContent = xmlBuilder.createTextNode(str(mathData))
            xmin.appendChild(xminContent)
            bndbox.appendChild(xmin)  # xmin標(biāo)簽結(jié)束
            ymin = xmlBuilder.createElement("ymin")  # ymin標(biāo)簽
            mathData = int(((float(oneline[2])) * Pheight + 1) - (float(oneline[4])) * 0.5 * Pheight)
            yminContent = xmlBuilder.createTextNode(str(mathData))
            ymin.appendChild(yminContent)
            bndbox.appendChild(ymin)  # ymin標(biāo)簽結(jié)束
            xmax = xmlBuilder.createElement("xmax")  # xmax標(biāo)簽
            mathData = int(((float(oneline[1])) * Pwidth + 1) + (float(oneline[3])) * 0.5 * Pwidth)
            xmaxContent = xmlBuilder.createTextNode(str(mathData))
            xmax.appendChild(xmaxContent)
            bndbox.appendChild(xmax)  # xmax標(biāo)簽結(jié)束
            ymax = xmlBuilder.createElement("ymax")  # ymax標(biāo)簽
            mathData = int(((float(oneline[2])) * Pheight + 1) + (float(oneline[4])) * 0.5 * Pheight)
            ymaxContent = xmlBuilder.createTextNode(str(mathData))
            ymax.appendChild(ymaxContent)
            bndbox.appendChild(ymax)  # ymax標(biāo)簽結(jié)束
            object.appendChild(bndbox)  # bndbox標(biāo)簽結(jié)束
            annotation.appendChild(object)  # object標(biāo)簽結(jié)束
        f = open(xmlPath + name[0:-4] + ".xml", 'w')
        xmlBuilder.writexml(f, indent='\t', newl='\n', addindent='\t', encoding='utf-8')
        f.close()
if __name__ == "__main__":
    picPath = "dataset/image/"  # 圖片所在文件夾路徑，后面的/一定要帶上
    txtPath = "dataset/label/"  # txt所在文件夾路徑，后面的/一定要帶上
    xmlPath = "dataset/Annotations/"  # xml文件保存路徑，后面的/一定要帶上
    makexml(picPath, txtPath, xmlPath)

activate yolo

labelimg image