วิธีตรวจสอบว่าไฟล์เป็นไฟล์รูปภาพที่ถูกต้องหรือไม่?

ฉันกำลังใช้ PIL

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

อย่างไรก็ตามแม้ว่าจะครอบคลุมกรณีส่วนใหญ่อย่างเพียงพอ แต่ก็ไม่พบไฟล์ภาพบางไฟล์เช่น xcf, svg และ psd ไฟล์ Psd แสดงข้อยกเว้น OverflowError

มีบางครั้งที่ฉันสามารถรวมไว้ด้วยได้หรือไม่?

หลายครั้งที่ตัวอักษรคู่แรกจะเป็นตัวเลขมหัศจรรย์สำหรับไฟล์รูปแบบต่างๆ คุณสามารถตรวจสอบสิ่งนี้เพิ่มเติมจากการตรวจสอบข้อยกเว้นด้านบน

ฉันเพิ่งพบโมดูลimghdr ในตัว จากเอกสาร python:

โมดูล imghdr กำหนดประเภทของรูปภาพที่อยู่ในไฟล์หรือสตรีมไบต์

นี่คือวิธีการทำงาน:

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

การใช้โมดูลนั้นดีกว่าการใช้ฟังก์ชันที่คล้ายกันซ้ำ

นอกจากสิ่งที่ Brian แนะนำแล้วคุณยังสามารถใช้วิธีการตรวจสอบของ PIL เพื่อตรวจสอบว่าไฟล์เสียหรือไม่

im.verify ()

พยายามตรวจสอบว่าไฟล์เสียหรือไม่โดยไม่ได้ถอดรหัสข้อมูลรูปภาพจริงๆ หากวิธีนี้พบปัญหาใด ๆ จะมีข้อยกเว้นที่เหมาะสม วิธีนี้ใช้ได้กับรูปภาพที่เพิ่งเปิดใหม่เท่านั้น หากโหลดรูปภาพแล้วผลลัพธ์จะไม่ได้กำหนด นอกจากนี้หากคุณต้องการโหลดภาพหลังจากใช้วิธีนี้คุณต้องเปิดไฟล์รูปภาพอีกครั้ง คุณลักษณะ

นอกจากการPILตรวจสอบรูปภาพแล้วคุณยังสามารถเพิ่มการตรวจสอบนามสกุลไฟล์ได้เช่นนี้:

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

โปรดทราบว่าสิ่งนี้จะตรวจสอบว่าชื่อไฟล์มีนามสกุลรูปภาพที่ถูกต้องหรือไม่ แต่ไม่ได้เปิดภาพเพื่อดูว่าเป็นภาพที่ถูกต้องหรือไม่นั่นเป็นเหตุผลที่คุณต้องใช้เพิ่มเติมPILหรือไลบรารีใด ๆ ที่แนะนำในคำตอบอื่น ๆ

อัปเดต

ฉันยังดำเนินการแก้ปัญหาต่อไปนี้ในสคริปต์ Python ของฉันที่นี่ใน GitHub

ฉันยังตรวจสอบด้วยว่าไฟล์ที่เสียหาย (jpg) มักไม่ใช่ภาพที่ 'เสีย' เช่นไฟล์รูปภาพที่เสียหายบางครั้งยังคงเป็นไฟล์รูปภาพที่ถูกต้องรูปภาพต้นฉบับสูญหายหรือมีการเปลี่ยนแปลง แต่คุณยังสามารถโหลดได้โดยไม่มีข้อผิดพลาด แต่การตัดไฟล์ทำให้เกิดข้อผิดพลาดเสมอ

สิ้นสุดการอัปเดต

คุณสามารถใช้โมดูลPython Pillow (PIL) กับรูปแบบรูปภาพส่วนใหญ่เพื่อตรวจสอบว่าไฟล์นั้นเป็นไฟล์รูปภาพที่ถูกต้องหรือไม่

ในกรณีที่คุณมุ่งเป้าไปที่การตรวจจับภาพที่แตกเช่นกัน @Nadia Alramli แนะนำim.verify()วิธีการนี้อย่างถูกต้องแต่ไม่พบข้อบกพร่องของภาพที่เป็นไปได้ทั้งหมดเช่นim.verifyตรวจไม่พบภาพที่ถูกตัดทอน (ซึ่งผู้ชมส่วนใหญ่มักจะโหลดด้วยพื้นที่สีเทา)

หมอนสามารถตรวจจับข้อบกพร่องประเภทนี้ได้เช่นกัน แต่คุณต้องใช้การปรับแต่งภาพหรือการถอดรหัส / การเข้ารหัสภาพหรือเพื่อเรียกใช้การตรวจสอบ สุดท้ายฉันขอแนะนำให้ใช้รหัสนี้:

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 เท่า (และฉันคิดว่าการพลิกเป็นหนึ่งในการแปลงที่ถูกกว่า) ด้วยรหัสนี้คุณจะตรวจสอบชุดภาพที่ความเร็วประมาณ 10 MBytes / วินาทีด้วย Pillow มาตรฐานหรือ 40 MBytes / วินาทีพร้อมโมดูล Pillow-SIMD (CPU 2.5Ghz x86_64 ที่ทันสมัย)

สำหรับรูปแบบอื่น ๆpsd , xcf , .. คุณสามารถใช้Imagemagick wrapper 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

บน Linux คุณสามารถใช้ python-magic ( http://pypi.python.org/pypi/python-magic/0.1 ) ซึ่งใช้ libmagic เพื่อระบุรูปแบบไฟล์

AFAIK, libmagic จะตรวจสอบไฟล์และพยายามบอกคุณเกี่ยวกับมันมากกว่าแค่รูปแบบเช่นขนาดบิตแมปเวอร์ชันรูปแบบ ฯลฯ ดังนั้นคุณอาจเห็นว่านี่เป็นการทดสอบ "ความถูกต้อง" เพียงผิวเผิน

สำหรับคำจำกัดความอื่น ๆ ของ "ถูกต้อง" คุณอาจต้องเขียนแบบทดสอบของคุณเอง

คุณสามารถใช้การผูก Python กับ libmagic, python-magicจากนั้นตรวจสอบประเภท mime สิ่งนี้จะไม่บอกคุณว่าไฟล์เสียหายหรือไม่เสียหาย แต่ควรจะสามารถระบุได้ว่าเป็นภาพประเภทใด

ฉันไม่รู้เกี่ยวกับด้านในของ psd แต่ฉันแน่ใจว่าแท้จริงแล้ว svg ไม่ใช่ไฟล์ภาพต่อ se - มันขึ้นอยู่กับ xml ดังนั้นโดยพื้นฐานแล้วคือ a ไฟล์ข้อความธรรมดา

ทางเลือกหนึ่งคือการใช้filetypeแพ็คเกจ

การติดตั้ง

python -m pip install filetype

ข้อดี

1. รวดเร็ว: ทำงานได้โดยการโหลดรูปภาพสองสามไบต์แรก ( ตรวจสอบหมายเลขวิเศษ )
2. รองรับประเภทละครใบ้ที่แตกต่างกัน: รูปภาพ, วิดีโอ, แบบอักษร, เสียง, ที่เก็บถาวร

ตัวอย่างโซลูชัน

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

การตรวจสอบนามสกุลไฟล์เป็นสิ่งที่ยอมรับได้หรือคุณกำลังพยายามยืนยันว่าข้อมูลนั้นเป็นไฟล์รูปภาพหรือไม่

หากคุณสามารถตรวจสอบนามสกุลของไฟล์ได้ว่านิพจน์ทั่วไปหรือการเปรียบเทียบแบบธรรมดาสามารถตอบสนองความต้องการได้

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)