参考代码

import imghdr
import os
imgType_list = {'jpg','bmp','png','jpeg','rgb','tif'}
if imghdr.what(item) in imgType_list:
    print('{} is picture file'.format(item))

REF

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)

https://www.cnblogs.com/wangshx666/p/14329137.html