Python Pillow - 图像文件识别

使用 Python Pillow (PIL) 库识别图像文件涉及检查文件扩展名以确定文件是否可能是图像。虽然 Pillow 本身没有内置方法来识别图像文件，但我们可以使用 **os** 模块的函数，例如 **listdir()、path.splitext()、path.is_image_file()、path.join()**，根据用户的需求根据扩展名过滤文件。

Python 的 **os.path** 模块没有提供直接的 **is_image_file()** 函数来识别图像文件。相反，我们通常可以使用 Pillow 库 (PIL) 来检查文件是否为图像。

但是，我们可以使用 **os.path** 并结合 Pillow 创建一个自定义的 **is_image_file()** 函数，根据文件的扩展名来检查文件是否似乎是图像。

以下是如何使用 **os.path** 检查文件扩展名并确定文件是否可能是图像文件的示例：

• 我们定义了 **is_image_file()** 函数，它首先检查文件扩展名是否看起来是常见的图像格式（在“image_extensions”中定义）。如果文件扩展名未被识别，则假定它不是图像文件。

• 对于识别的图像文件扩展名，该函数尝试使用 Pillow 的 **Image.open()** 将文件作为图像打开。如果此操作成功且没有错误，则该函数返回 **True**，表示该文件是有效的图像。

• 如果打开文件时出现问题，则该函数返回 **False**。

• 此方法结合了 **os.path** 用于提取文件扩展名，以及 Pillow 用于检查文件是否为有效图像。

示例

import os
from PIL import Image
def is_image_file(file_path):
   image_extensions = {".jpg", ".jpeg", ".png", ".gif", ".bmp", ".tiff", ".ico"}  
#Add more extensions as needed
   _, file_extension = os.path.splitext(file_path)

#Check if the file extension is in the set of image extensions
    return file_extension.lower() in image_extensions
file_path = "Images/butterfly.jpg"
if is_image_file(file_path):
   print("This is an image file.")
   Image.open(file_path)
else:	
   print("This is not an image file.")

输出

This is an image file.

示例

在此示例中，我们传递文件扩展名为 **pdf**，因为它不是图像文件扩展名，因此结果将不是图像文件。

import os
from PIL import Image
def is_image_file(file_path):
   image_extensions = {".jpg", ".jpeg", ".png", ".gif", ".bmp", ".tiff", ".ico"}  
#Add more extensions as needed
   _, file_extension = os.path.splitext(file_path)

#Check if the file extension is in the set of image extensions
   return file_extension.lower() in image_extensions
file_path = "Images/butterfly.pdf"
if is_image_file(file_path):
   print("This is an image file.")
   Image.open(file_path)
else:
   print("This is not an image file.")

输出

This is not an image file.