Python推测图片格式

Python推测图片格式 使用imghdr模块

imghdr模块提供了what()方法，该方法可以推测文件或字节流中的图像的类型。可以识别的图片格式见Python文档。为防止打不开，我把截图放在下面。

imghdr的使用方法比较简单，如下所示。你可以传入一个文件，也可以传入字节流。

import imghdr


# 文件形式
res = imghdr.what("/home/zy/图片/渴望.png")     
print(res)


# 字节流形式
import requests
data = requests.get(url="https://gimg2.baidu.com/image_search/src=http://www.kaotop.com/skin/sinaskin/image/nopic.gif")
res = imghdr.what(None, data.content)       # data.content是bytes类型。
print(res) 

执行结果如下所示：

png
jpeg