python imghdr判断文件是否是图片

1、listdir函数可以遍历读取文件夹下的文件。
2、imghdr 可以用来判断文件是否是图片

import imghdr
import os

imgType_list = {'jpg','bmp','png','jpeg','rgb','tif'}

for item in os.listdir(file_name): 
     # imghdr 可以用来判断文件是否是图片
    if imghdr.what(item) in imgType_list:
        print(item)