转录自https://blog.csdn.net/u010167269/article/details/51084312,感谢大神分享。今天在做目标检测,我发现图片数量大于标注文件数量,就找到这个小程序,完美运行!
# !/usr/bin/env python
# encoding: utf-8
import os
import glob
from PIL import Image
#指定找到文件后,另存为的文件夹路径
outDir = os.path.abspath('F:/Graduate_data/目标检测/Catenary-5000/output')
#指定第一个文件夹的位置
imageDir1 = os.path.abspath('F:/Graduate_data/目标检测/Catenary-5000/Catenary-5000/JPEGImages')
#定义要处理的第一个文件夹变量
image1 = [] #image1指文件夹里的文件,包括文件后缀格式;
imgname1 = [] #imgname1指里面的文件名称,不包括文件后缀格式
#通过glob.glob来获取第一个文件夹下,所有'.jpg'文件
imageList1 = glob.glob(os.path.join(imageDir1, '*.jpg'))
#遍历所有文件,获取文件名称(包括后缀)
for item in imageList1:
image1.append(os.path.basename(item))
#遍历文件名称,去除后缀,只保留名称
for item in image1:
(temp1, temp2) = os.path.splitext(item)
imgname1.append(temp1)
#对于第二个文件夹路径,做同样的操作