一行代码搞定
获取文件夹下面的所有xml文件
import glob
train_path = 'D:\\AI_Kaggle_Study\voc2coco\\mask\\train\\'
xml_list = glob.glob(train_path + "/*.xml")
print(xml_list)
或者
import os
xml_list = [os.path.join(train_path, i) for i in os.listdir(train_path) if i[-4:] == '.xml']
print(xml_list )
或者
import os
xml_list = [os.path.join(train_path, i) for i in os.listdir(train_path) if i.endswith('.xml')]
print(xml_list )
或者
import os
xml_list = [os.path.join(train_path, i) for i in os.listdir(train_path) if os.path.splitext(i)[-1] == '.xml']
print(xml_list)
注意:上面的xml_list
列表存储的是所有的完整的文件名路径