import os
import sys
import numpy as np
from scipy.misc import imread
dir = '/data/VOCdevkit/VOC2007/JPEGImages'#指定图片所在的文件夹,该文件夹下存放的文件全为图片
imglist = os.listdir(dir)#imglist为dir文件夹下的所有文件名
num_files = len(imglist)
count = 0#用来统计不是三通道的图片数目
for x in range(0,num_files-1):#遍历dir下所有的图片
file_path = os.path.join(dir, imglist[x])#拼接某一图片的路径,imglist[x]为文件名
im_in = np.array(imread(file_path))
if len(im_in.shape) == 2:
im_in = im_in[:, :, np.newaxis]
im_in = np.concatenate((im_in, im_in, im_in), axis=2)
im_in = im_in[:, :, ::-1]
im = im_in
if im.shape[2] != 3:#若图片不是三通道
count += 1
print(imglist[x])
print(count)