Python用imghdr模块识别图片格式实例解析
imghdr模块
功能描述：imghdr模块用于识别图片的格式。它通过检测文件的前几个字节，从而判断图片的格式。
唯一一个API
imghdr.what(file,h=None)
第一个参数file可以是用rb模式打开的file对象或者表示路径的字符串和PathLike对象。h参数是一段字节串。函数返回表示图片格式的字符串。
>>>importimghdr
>>>imghdr.what('test.jpg')
'jpeg'
具体的返回值和描述如下：
| 返回值 | 描述 | 检测方式 |
|---|---|---|
| jpeg | 用JFIF或者Exif格式保存的JPEG图片 | 第7到第10个字节是b'JFIF'或者b'Exif' |
| png | 可移植网络图形格式(PortableNetworkGraphicFormat) | 以字节串b'\x89PNG\r\n\x1a\n'开头 |
| gif | GIF(GraphicsInterchangeFormat)的87版本和89版本 | 前6个字节为b'GIF87a'或者b'GIF89a' |
| tiff | TIFF(TagImageFileFormat)的两种字节顺序 | 前两个字节为b'MM'或者b'II' |
| rgb | SGIImgLib | 以字节串b'\x01\xda'开头 |
| pbm | PortableBitmap | 第1个字节为b'P'，第2个字节为b'1'或b'4'，第3个字节为b'\t'或b'\n'或b'\r' |
| pgm | PortableGraymapFiles | 第1个字节为b'P'，第2个字节为b'2'或b'5'，第3个字节为b'\t'或b'\n'或b'\r' |
| ppm | PortablePixmapFiles | 第1个字节为b'P'，第2个字节为b'3'或b'6'，第3个字节为b'\t'或b'\n'或b'\r' |
| rast | SunRaster | 以字节串b'\x59\xA6\x6A\x95'开头 |
| xbm | XBitmapFiles | 以字节串b'#define‘开头 |
| bmp | Bitmap，Windows标准图像文件格式 | 以字节串b'BM'开头 |
| webp | 谷歌的WebP格式，Python3.5加入 | 以字节串b'RIFF'开头并且第9到第12个字节为b'WEBP' |
| exr | OpenEXR，Python3.5加入 | 以字节串b'\x76\x2f\x31\x01'开头 |
模块内部缺陷
当h参数不为空时，模块会忽略掉file参数，直接检测h参数，但此时file参数又是必须提供的，算是一个设计缺陷吧。博主个人感觉这个h参数根本没有存在的意义，没必要放在参数列表里面。
>>>importimghdr
>>>imghdr.what('test.jpg',b'\x89PNG\r\n\x1a\n')
'png'
>>>
自定义检测流程
imghdr内部使用了test_jpeg、test_png、test_gif等函数检测文件的格式。模块内部维护了一个函数列表imghdr.tests，每次调用what函数的时候，会按列表里的顺序调用检测函数，当检测函数返回结果时退出循环。用户可以通过修改这个列表达到修改检测流程的目的。同时，也可以自行添加检测函数到列表里面。
下面这个例子，博主添加了一个函数放在检测流程的最后提示文件不是图片：
>>>importimghdr
>>>deffinal(h,f):
...print("Thisfileisn\'taimage!")
...
>>>imghdr.tests.append(final)
>>>imghdr.what("imghdr.md")
Thisfileisn'taimage!
自行添加检测函数需要接收两个参数h和f，h是用来检测的字节串，f是file对象。但是在模块内部自带的检测函数都是没用到这个f参数的…
命令行启动imghdr模块
博主读源码的时候发现了官方文档里面没提到的两个函数，提供了直接用命令行启动imghdr模块的方法。
只需要调用python-mimghdr[-r]file1file2...就可以直接检测文件的类型。file可以是文件也可以是文件夹。该命令默认只检测文件夹下一层的文件类型，如果需要递归检测，则加上-r参数。
每个文件以”文件名:文件类型/None”的形式打印一行输出结果。
总结：可能是因为使用的人少的原因，imghdr这个模块内部有挺多小问题，不过只要老老实实按照官方提供的文档使用是不会出事的。博主后面试着改一下这个模块然后pr上去。
以上就是本文关于Python用imghdr模块识别图片格式实例解析的全部内容，希望对大家有所帮助。感兴趣的朋友可以继续参阅本站其他相关专题，如有不足之处，欢迎留言指出。感谢朋友们对本站的支持！
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。