Suj*_*joy 89 python image identification imghdr
我目前正在使用PIL.
from PIL import Image
try:
im=Image.open(filename)
# do stuff
except IOError:
# filename not an image file
但是,尽管这足以涵盖大多数情况,但未检测到某些图像文件,如xcf,svg和psd.Psd文件抛出OverflowError异常.
有没有我可以包括他们？
Nad*_*mli 175
我刚刚找到了内置的imghdr模块.从python文档:
imghdr模块确定文件或字节流中包含的图像类型.
这是它的工作原理:
>>> import imghdr
>>> imghdr.what('/tmp/bass')
'gif'
使用模块比重新实现类似功能要好得多
Nad*_*mli 45
除了Brian建议您可以使用PIL的验证方法来检查文件是否损坏.
im.verify()
尝试确定文件是否已损坏,而不实际解码图像数据.如果此方法发现任何问题,则会引发适当的异常.此方法仅适用于新打开的图像; 如果图像已加载,则结果未定义.此外,如果需要在使用此方法后加载图像,则必须重新打开图像文件.属性
tsv*_*iko 17
除了PIL图像检查之外，您还可以像这样添加文件扩展名检查：
filename.lower().endswith(('.png', '.jpg', '.jpeg', '.tiff', '.bmp', '.gif'))
请注意，这仅检查文件名是否具有有效的图像扩展名，它实际上并未打开图像以查看它是否为有效图像，这就是您需要额外使用PIL或其他答案中建议的库之一的原因。
Bri*_*ndy 10
很多时候,前几个字符将是各种文件格式的神奇数字.除了上面的异常检查之外,您还可以检查这一点.
更新
我还在GitHub 上的Python 脚本中实现了以下解决方案。
我还验证了损坏的文件 (jpg) 经常不是“损坏的”图像，即损坏的图片文件有时仍然是合法的图片文件，原始图像丢失或更改但您仍然可以正确加载它。但是，文件截断总是会导致错误。
结束更新
您可以使用 Python Pillow (PIL) 模块和大多数图像格式来检查文件是否是有效且完整的图像文件。
如果您的目标是检测损坏的图像，@Nadia Alramli 正确地建议了该im.verify()方法，但这并不能检测所有可能的图像缺陷，例如，im.verify不能检测截断的图像（大多数观众经常加载灰色区域）。
Pillow也能够检测这些类型的缺陷，但您必须应用图像处理或图像解码/重新编码或触发检查。最后我建议使用这个代码：
try:
im = Image.load(filename)
im.verify() #I perform also verify, don't know if he sees other types o defects
im.close() #reload is necessary in my case
im = Image.load(filename)
im.transpose(PIL.Image.FLIP_LEFT_RIGHT)
im.close()
except:
#manage excetions here
在图像缺陷的情况下，此代码将引发异常。请考虑 im.verify 比执行图像处理快 100 倍（我认为翻转是更便宜的转换之一）。使用此代码，您将使用标准 Pillow 以约 10 MBytes/sec 或使用 Pillow-SIMD 模块（现代 2.5Ghz x86_64 CPU）以 40 MBytes/sec 验证一组图像。
对于其他格式psd , xcf ,.. 可以使用Imagemagick包装器Wand，代码如下：
im = wand.image.Image(filename=filename)
temp = im.flip;
im.close()
但是，根据我的实验，Wand 没有检测到截断的图像，我认为它会在没有提示的情况下将缺少的部分加载为灰色区域。
我红认为，ImageMagick的具有外部命令识别是可以做的工作，但我还没有找到一种方法来编程方式调用该函数，我没有测试过这条路线。
我建议始终执行初步检查，检查文件大小不为零（或非常小），这是一个非常便宜的想法：
statfile = os.stat(filename)
filesize = statfile.st_size
if filesize == 0:
#manage here the 'faulty image' case
一种选择是使用filetype包。
安装
python -m pip install filetype
好处
例子
文件类型 >= 1.0.7
import filetype
filename = "/path/to/file.jpg"
if filetype.is_image(filename):
print(f"{filename} is a valid image...")
elif filetype.is_video(filename):
print(f"{filename} is a valid video...")
文件类型 <= 1.0.6
import filetype
filename = "/path/to/file.jpg"
if filetype.image(filename):
print(f"{filename} is a valid image...")
elif filetype.video(filename):
print(f"{filename} is a valid video...")
官方 repo 的附加信息：https : //github.com/h2non/filetype.py
在 Linux 上，您可以使用python-magic，它使用libmagic来识别文件格式。
AFAIK，libmagic 查看文件并试图告诉你更多关于它的信息，而不仅仅是格式，比如位图尺寸、格式版本等。所以你可能会认为这是对“有效性”的表面测试。
对于“有效”的其他定义，您可能必须编写自己的测试。