删除VOC数据中没打标签的图片
py脚本功能:删除VOC数据中的JPEGImages文件中的没打标签的文件。
代码如下:
import os
path1 = r"F:\alldate\SHONG\Annotations" #对比的文件夹(标签文件夹)
path2 = r"F:\alldate\SHONG\JPEGImages" #删除文件的文件夹(图片文件夹)
filelist1 = os.listdir(path1) #该文件夹下所有的文件(包括文件夹)
filelist2 = os.listdir(path2)
tp = ".xml" #要匹配标签文件的后缀
for file2 in filelist2:
filename = os.path.splitext(file2)[0] + tp # 匹配文件名
if filename not in filelist1: #如果没有对应匹配文件,就删除
os.remove(os.path.join(path2,file2))