我正在嘗試訓練一個新的 Tensorflow 模型,并且需要將 .xml 文件聚合到一個 .csv 中。我找到了一個腳本來執行此操作,但它使用 glob 模塊——我不熟悉的東西。我正在尋求幫助以了解此代碼的工作原理,以便我可以適當地對其進行調整。我正在從一個目錄執行 .py,該目錄包含兩個帶有 .jpg 和 .xml 文件的子目錄。import osimport globimport pandas as pdimport xml.etree.ElementTree as ETdef xml_to_csv(path):xml_list = []for xml_file in glob.glob(path + '/*.xml'): tree = ET.parse(xml_file) root = tree.getroot() for member in root.findall('object'): value = (root.find('filename').text, int(root.find('size')[0].text), int(root.find('size')[1].text), member[0].text, int(member[4][0].text), int(member[4][1].text), int(member[4][2].text), int(member[4][3].text) ) xml_list.append(value)column_name = ['filename', 'width', 'height', 'class', 'xmin', 'ymin', 'xmax', 'ymax']xml_df = pd.DataFrame(xml_list, columns=column_name)return xml_dfdef main(): image_path = os.path.join(os.getcwd(), 'annotations') xml_df = xml_to_csv(image_path) xml_df.to_csv('raccoon_labels.csv', index=None) print('Successfully converted xml to csv.')main()運行代碼沒有問題 - 它創建 .csv 文件,但由于標題下沒有數據,我假設它沒有找到任何 .xml 文件。
2 回答

四季花海
TA貢獻1811條經驗 獲得超5個贊
發現問題了!需要重新定向“注釋”文件夾,這解決了它。
此外,找到了這個腳本的更好版本:https ://github.com/EdjeElectronics/TensorFlow-Object-Detection-API-Tutorial-Train-Multiple-Objects-Windows-10/blob/master/xml_to_csv.py
添加回答
舉報
0/150
提交
取消