因項(xiàng)目需要根據(jù)指定格式的文件生成XML標(biāo)注文件，可以方便使用LabelImg打開進(jìn)行編輯和查看。其原始文件默認(rèn)使用逗號(hào)進(jìn)行分隔，如下所示：

第1個(gè)值：原始圖片中切圖小文件，以AIpng_x，其中x代表原始圖片的第幾個(gè)切圖文件 第2~5值：分別對(duì)應(yīng)于ymin, xmin, ymax, xmax 第6個(gè)值：代表對(duì)應(yīng)的標(biāo)簽標(biāo)注

    在生成XML文件時(shí)，需要對(duì)其進(jìn)行匯總，即將屬于同一個(gè)原始文件的切圖小文件的標(biāo)注匯總到一起，其實(shí)現(xiàn)代碼如下所示：

import osfrom Logger import MyLoggerfrom xml.dom.minidom import Documentfrom collections import defaultdictimport reclass OpeateXML: def __init__(self, srcPath: str, targetPath: str, srcFileName: str): self._srcPath = srcPath self._targetPath = targetPath self._srcFileName = srcFileName def readSrcFileName(self, fileEncoding='utf8') -> dict: data = defaultdict(list) s = re.compile('.AIpng_d{1,}', re.IGNORECASE) srcFileFullPath = os.path.join(self._srcPath, self._srcFileName) try: with open(srcFileFullPath, mode='r', encoding=fileEncoding, errors='ignore') as fr:for content in fr.readlines(): data[s.sub('.AIpng',content.strip().split(',')[0])].append(content.strip()) except Exception as ex: MyLogger().error(f'OperateXML:read file error:n{ex}') return {} else: # data.sort(key=lambda x: x.strip().split(',')[0]) return data def createXML(self, data: dict, fileEncoding='utf8'): if data: try:for k,v in data.items(): doc = Document() # 創(chuàng)建根節(jié)點(diǎn) rootNode = doc.createElement('annotation') # 添加根節(jié)點(diǎn) doc.appendChild(rootNode) folder = doc.createElement('folder') folderText = doc.createTextNode(self._targetPath) folder.appendChild(folderText) rootNode.appendChild(folder) filename = doc.createElement('filename') filenameText = doc.createTextNode(k) filename.appendChild(filenameText) rootNode.appendChild(filename) path = doc.createElement('path') pathText = doc.createTextNode(os.path.join(self._targetPath,k)) path.appendChild(pathText) rootNode.appendChild(path) for i in v: tmpData = i.strip().split(',') if len(tmpData) == 6: _, ymin, xmin, ymax, xmax, labelName = tmpData objectObj = doc.createElement('object') rootNode.appendChild(objectObj) objectName = doc.createElement('name') objectNameText = doc.createTextNode(labelName) objectName.appendChild(objectNameText) objectObj.appendChild(objectName) objectBndBox = doc.createElement('bndbox') objectObj.appendChild(objectBndBox) objectBndBoxXmin = doc.createElement('xmin') objectBndBoxYmin = doc.createElement('ymin') objectBndBoxXmax = doc.createElement('xmax') objectBndBoxYmax = doc.createElement('ymax') objectBndBoxXminText = doc.createTextNode(xmin) objectBndBoxYminText = doc.createTextNode(ymin) objectBndBoxXmaxText = doc.createTextNode(xmax) objectBndBoxYmaxText = doc.createTextNode(ymax) objectBndBox.appendChild(objectBndBoxXmin) objectBndBox.appendChild(objectBndBoxYmin) objectBndBox.appendChild(objectBndBoxXmax) objectBndBox.appendChild(objectBndBoxYmax) objectBndBoxXmin.appendChild(objectBndBoxXminText) objectBndBoxYmin.appendChild(objectBndBoxYminText) objectBndBoxXmax.appendChild(objectBndBoxXmaxText) objectBndBoxYmax.appendChild(objectBndBoxYmaxText) objectObj.appendChild(objectBndBox) else: continue # save xml xmlName=os.path.splitext(k)[0]+'.xml' targetPath = os.path.join(self._targetPath, xmlName) with open(targetPath, mode='w', encoding=fileEncoding) as fw: doc.writexml(fw, indent='t', newl='n', addindent='t', encoding=fileEncoding) except Exception as ex:MyLogger().error(f'OperateXML:Save xml errorn{ex}')returnif __name__ == ’__main__’: srcPath = r'C:UsersSurpassDocumentsPycharmProjectsdataTEST-8outs' srcName = 'locations.txt' targetPath = r'C:UsersSurpassDocumentsPycharmProjectsdataTEST-8outsin_number' operateXML = OpeateXML(srcPath, targetPath, srcName) a = operateXML.readSrcFileName() operateXML.createXML(a)

    最終生成的XML效果如下所示：

    在LabelImg中的效果如下所示：

PS：這里再為大家提供幾款關(guān)于xml操作的在線工具供大家參考使用：

在線XML/JSON互相轉(zhuǎn)換工具：http://tools.jb51.net/code/xmljson

在線格式化XML/在線壓縮XML：http://tools.jb51.net/code/xmlformat

XML在線壓縮/格式化工具：http://tools.jb51.net/code/xml_format_compress

XML代碼在線格式化美化工具：http://tools.jb51.net/code/xmlcodeformat

對(duì)關(guān)于Python生成XML相關(guān)內(nèi)容感興趣的讀者可查看本站專題：《Python操作xml數(shù)據(jù)技巧總結(jié)》

以上就是Python根據(jù)指定文件生成XML的方法的詳細(xì)內(nèi)容，更多關(guān)于Python生成XML的資料請(qǐng)關(guān)注好吧啦網(wǎng)其它相關(guān)文章！