判断文件夹中是否全是jpg图片

import numpy as np
import os
train_dir = '/media/chenyu/6046436546433AD4/48 class hand/VOC2007/JPEGImages/'
def get_files(file_dir):
    name = []
    for file in os.listdir(file_dir):
            name.append(file_dir + file)
    print('There are %d image' %(len(name)))
    return name
    
a = get_files(train_dir)
b = len(a)
from PIL import Image
def is_jpg(filename):
    try:
        i=Image.open(filename)
        return i.format =='JPEG'
    except IOError:
        print(filename)
        print('***********************************************************************')
        return False
for i in range(b):
    is_jpg(a[i])
    #print(a[i])