Python图像处理中常用的imghdr模块介绍

imghdr模块是Python中用来检查图像文件格式的标准库之一。它可以读取文件的几个字节来判断文件的格式，而不是通过文件扩展名来判断。这个模块非常有用，特别是处理大量或未知文件格式的图像时。

imghdr模块提供了一个函数：imghdr.what(filename, h=None)，它接受一个文件名作为参数，并尝试使用不同的方法来确定文件的格式。根据文件的前几个字节，它可以识别的文件格式包括：BMP、JPEG、GIF、PNG和PPM。

下面是一个使用imghdr模块的简单示例：

import imghdr

def check_image_format(filename):
    format = imghdr.what(filename)
    if format is None:
        print("Unknown format")
    else:
        print("The format of the image is: {}".format(format))

check_image_format("image.jpg")

在上面的例子中，我们使用imghdr模块的what()函数来检查名为image.jpg的图像文件的格式。如果该文件的格式是可识别的，则在屏幕上显示格式名称。否则，显示"Unknown format"。

除了传递文件名，what()函数还可以传递一个字节串（由文件的前几个字节组成）作为可选参数h。这种情况下，它将使用给定的字节串来检查文件格式。例如：

import imghdr

def check_image_format_with_bytes(bytes):
    format = imghdr.what("", h=bytes)
    if format is None:
        print("Unknown format")
    else:
        print("The format of the image is: {}".format(format))

with open("image.jpg", "rb") as file:
    bytes = file.read(32)
    check_image_format_with_bytes(bytes)

在上面的例子中，我们首先读取名为image.jpg的图像文件的前32个字节，并将它们传递给what()函数来检查格式。

总之，imghdr模块是一个很方便的工具，用于确定图像文件的实际格式，而不依赖于文件扩展名。它可以帮助我们更有效地处理大量或未知格式的图像文件。