Làm thế nào để kiểm tra xem một tệp có phải là tệp hình ảnh hợp lệ hay không?

Tôi hiện đang sử dụng PIL.

from PIL import Image
try:
    im=Image.open(filename)
    # do stuff
except IOError:
    # filename not an image file

Tuy nhiên, trong khi điều này đủ bao gồm hầu hết các trường hợp, một số tệp hình ảnh như, xcf, svg và psd không được phát hiện. Tệp psd ném một ngoại lệ OverflowError.

Có cách nào đó mà tôi có thể bao gồm chúng không?

Rất nhiều lần cặp ký tự đầu tiên sẽ là một con số kỳ diệu cho các định dạng tệp khác nhau. Bạn có thể kiểm tra điều này ngoài việc kiểm tra ngoại lệ của bạn ở trên.

Tôi vừa tìm thấy mô-đun imghdr nội trang . Từ tài liệu python:

Mô-đun imghdr xác định loại hình ảnh có trong tệp hoặc luồng byte.

Đây là cách nó hoạt động:

>>> import imghdr
>>> imghdr.what('/tmp/bass')
'gif'

Sử dụng một mô-đun tốt hơn nhiều so với việc thực hiện lại chức năng tương tự

Ngoài những gì Brian đang đề xuất, bạn có thể sử dụng phương pháp xác minh của PIL để kiểm tra xem tệp có bị hỏng hay không.

im.verify ()

Cố gắng xác định xem tệp có bị hỏng hay không mà không thực sự giải mã dữ liệu hình ảnh. Nếu phương pháp này tìm thấy bất kỳ vấn đề nào, nó sẽ đưa ra các ngoại lệ phù hợp. Phương pháp này chỉ hoạt động trên một hình ảnh mới mở; nếu hình ảnh đã được tải, kết quả là không xác định. Ngoài ra, nếu bạn cần tải hình ảnh sau khi sử dụng phương pháp này, bạn phải mở lại tệp hình ảnh. Thuộc tính

Ngoài PILkiểm tra hình ảnh, bạn cũng có thể thêm kiểm tra phần mở rộng tên tệp như sau:

filename.lower().endswith(('.png', '.jpg', '.jpeg', '.tiff', '.bmp', '.gif'))

Lưu ý rằng điều này chỉ kiểm tra xem tên tệp có phần mở rộng hình ảnh hợp lệ hay không, nó không thực sự mở hình ảnh để xem đó có phải là hình ảnh hợp lệ hay không, đó là lý do tại sao bạn cần sử dụng bổ sung PILhoặc một trong các thư viện được đề xuất trong các câu trả lời khác.

Cập nhật

Tôi cũng đã triển khai giải pháp sau trong tập lệnh Python của mình tại đây trên GitHub .

Tôi cũng xác minh rằng các tệp bị hỏng (jpg) thường không phải là ảnh 'bị hỏng', tức là tệp ảnh bị hỏng đôi khi vẫn là tệp ảnh hợp pháp, ảnh gốc bị mất hoặc bị thay đổi nhưng bạn vẫn có thể tải nó mà không bị lỗi. Tuy nhiên, việc cắt ngắn tệp luôn gây ra lỗi.

Kết thúc cập nhật

Bạn có thể sử dụng mô-đun Python Pillow (PIL), với hầu hết các định dạng hình ảnh, để kiểm tra xem tệp có phải là tệp hình ảnh hợp lệ và nguyên vẹn hay không.

Trong trường hợp bạn nhắm đến việc phát hiện cả hình ảnh bị hỏng, @Nadia Alramli đề xuất chính xác im.verify()phương pháp, nhưng điều này không phát hiện tất cả các lỗi hình ảnh có thể có , ví dụ: im.verifykhông phát hiện hình ảnh bị cắt ngắn (mà hầu hết người xem thường tải với một vùng xám).

Pillow cũng có thể phát hiện các loại khuyết tật này, nhưng bạn phải áp dụng thao tác hình ảnh hoặc giải mã / mã hóa hình ảnh trong hoặc để kích hoạt kiểm tra. Cuối cùng, tôi đề nghị sử dụng mã này:

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

Trong trường hợp có lỗi hình ảnh, mã này sẽ đưa ra một ngoại lệ. Hãy coi rằng im.verify nhanh hơn khoảng 100 lần so với việc thực hiện thao tác trên hình ảnh (và tôi nghĩ rằng lật là một trong những phép biến đổi rẻ hơn). Với mã này, bạn sẽ xác minh một tập hợp hình ảnh ở tốc độ khoảng 10 MByte / giây với Pillow tiêu chuẩn hoặc 40 MByte / giây với mô-đun Pillow-SIMD (CPU 2,5Ghz x86_64 hiện đại).

Đối với các định dạng khác psd , xcf , .. bạn có thể sử dụng Imagemagick wrapper Wand , mã như sau:

im = wand.image.Image(filename=filename)
temp = im.flip;
im.close()

Tuy nhiên, từ các thử nghiệm của tôi, Wand không phát hiện ra hình ảnh bị cắt bớt, tôi nghĩ rằng nó tải các phần thiếu dưới dạng vùng xám mà không cần nhắc.

Tôi đỏ mà ImageMagick có một lệnh bên ngoài xác định rằng có thể làm cho công việc, nhưng tôi đã không tìm thấy một cách để gọi chức năng lập trình và tôi đã không kiểm tra tuyến đường này.

Tôi đề nghị luôn thực hiện kiểm tra sơ bộ, kiểm tra kích thước tệp để không bằng 0 (hoặc rất nhỏ), là một ý tưởng rất rẻ :

statfile = os.stat(filename)
filesize = statfile.st_size
if filesize == 0:
  #manage here the 'faulty image' case

Trên Linux, bạn có thể sử dụng python-magic ( http://pypi.python.org/pypi/python-magic/0.1 ) sử dụng libmagic để xác định định dạng tệp.

AFAIK, libmagic nhìn vào tệp và cố gắng cho bạn biết nhiều hơn về nó chứ không chỉ là định dạng, như kích thước bitmap, phiên bản định dạng, v.v. Vì vậy, bạn có thể xem đây là một bài kiểm tra bề ngoài về "tính hợp lệ".

Đối với các định nghĩa khác về "hợp lệ", bạn có thể phải viết các bài kiểm tra của riêng mình.

Bạn có thể sử dụng các liên kết Python với libmagic, python-magic và sau đó kiểm tra các loại kịch câm. Điều này sẽ không cho bạn biết nếu các tệp bị hỏng hay còn nguyên vẹn nhưng nó sẽ có thể xác định loại hình ảnh đó là.

Vâng, tôi không biết về bên trong của psd, nhưng tôi, chắc chắn, biết rằng, trên thực tế, svg không phải là một tệp hình ảnh, - nó dựa trên xml, vì vậy về cơ bản, nó là một tệp văn bản thuần túy.

Một lựa chọn là sử dụng filetypegói.

Cài đặt

python -m pip install filetype

Ưu điểm

1. Nhanh chóng: Nó có hoạt động bằng cách tải vài byte đầu tiên của hình ảnh của bạn ( kiểm tra con số kỳ diệu )
2. Hỗ trợ các loại kịch câm khác nhau: Hình ảnh, Video, Phông chữ, Âm thanh, Lưu trữ.

Ví dụ giải pháp

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...")

Thông tin bổ sung về repo chính thức: https://github.com/h2non/filetype.py

Việc kiểm tra phần mở rộng tệp có được chấp nhận không hay bạn đang cố gắng xác nhận rằng bản thân dữ liệu đại diện cho tệp hình ảnh?

Nếu bạn có thể kiểm tra phần mở rộng tệp, một biểu thức chính quy hoặc một phép so sánh đơn giản có thể đáp ứng yêu cầu.

format = [".jpg",".png",".jpeg"]
 for (path,dirs,files) in os.walk(path):
     for file in files:
         if file.endswith(tuple(format)):
             print(path)
             print ("Valid",file)
         else:
             print(path)
             print("InValid",file)