判断一个路径是不是图片格式，如果是的话再执行
import os
def is_image_file(file_path):
# 定义常见的图片文件扩展名
image_extensions = {'.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff', '.webp'}
# 获取文件的扩展名并判断是否在图片扩展名集合中
file_extension = os.path.splitext(file_path)[1].lower()
return file_extension in image_extensions
file_path = "/path/to/your/file.jpg"
if is_image_file(file_path):
print(f"{file_path} 是一个图片文件")
else:
print(f"{file_path} 不是一个图片文件")
也可以使用 try exception
(1)使用 PIL的try-except
from PIL import Image
file_path = "path/to/your/image.jpg"
try:
image_pil = Image.open(file_path)
image_pil.show()
except FileNotFoundError:
print("文件未找到，请检查路径是否正确。")
except Exception as e:
print(f"发生了一个错误：{e}")
(2)使用cv2的try-except
import cv2
file_path = "path/to/your/image.jpg"
try:
image_cv2 = cv2.imread(file_path)
if image_cv2 is None:
raise FileNotFoundError("文件未找到，请检查路径是否正确。")
cv2.imshow('Image', image_cv2)
cv2.waitKey(0)
cv2.destroyAllWindows()
except FileNotFoundError as e:
print(e)
except Exception as e:
print(f"发生了一个错误：{e}")
