亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

如何確保 glob 命令“找到”我需要聚合的文件?

如何確保 glob 命令“找到”我需要聚合的文件?

鴻蒙傳說 2022-06-22 15:41:48
我正在嘗試訓練一個新的 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 回答

?
慕碼人8056858

TA貢獻1803條經驗 獲得超6個贊

你的glob電話寫得不好。你需要讓它進入子文件夾

glob.glob(path + '**/*.xml')


查看完整回答
反對 回復 2022-06-22
?
四季花海

TA貢獻1811條經驗 獲得超5個贊

發現問題了!需要重新定向“注釋”文件夾,這解決了它。

此外,找到了這個腳本的更好版本:https ://github.com/EdjeElectronics/TensorFlow-Object-Detection-API-Tutorial-Train-Multiple-Objects-Windows-10/blob/master/xml_to_csv.py


查看完整回答
反對 回復 2022-06-22
  • 2 回答
  • 0 關注
  • 153 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號