在做训练集的时候,发现图片与相应的xml文件不匹配,写了一个脚本找出哪些图片没有标记
import os
files = os.listdir(".") #获取图片目路径
i = 0
filetype = ".jpg"#文件类型
for filename in files:
portion = os.path.splitext(filename)#将文件名拆成名字和后缀
if filetype == portion[1]:
if not os.path.exists(portion[0]+'.xml'):
print(portion[0])