在Python中使用 imghdr 确定图像的类型

简介

imghdr 是 Python 自带的一个用于确定图像文件类型的模块。我们可以利用 imghdr 模块来轻松地确定一幅图像的格式。

该模块适用于 JPEG、PNG、 GIF 和其他流行的图像格式。

安装

无需额外安装，imghdr 模块是 Python 自带的标准库。

使用方法

使用 imghdr 模块进行图片类型检测的方法如下：

import imghdr

img_type = imghdr.what("path/to/image.jpg")
print(img_type) # 'jpeg'

下面是一个利用 imghdr 模块辨别包含多种图片类型的文件的示例代码：

import imghdr

filename = 'path/to/image.file'

with open(filename, 'rb') as f:
    img_type = imghdr.what(f)

print(f'The image type is {img_type}')

参数说明

• imghdr.what(file[, h])：检查文件类型并返回字符串表示的图像类型。参数 file 可以是一个文件名字符串，也可以是一个文件对象。参数 h 是一个可选的文件头数据，用于确定文件类型，以 bytes 对象的形式传递。

支持的图片格式

imghdr 模块支持的图片格式如下：

• 'rgb', 'gif', 'pbm', 'pgm', 'ppm', 'tiff', 'rast', 'xbm', 'jpeg', 'bmp', 'png', 'webp', 'exr', 'tga', 'hdr', 'pic'

总结

在 Python 中使用 imghdr 确定图像类型是一个轻松且方便的方法。该模块不仅仅支持基本的图片格式，还支持各种流行的格式例如WebP等。