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)