Python作為人工智能的開發語言，也是很多人學習的方向，小猿圈加加分享一個Python知識點，希望幫助那些正在學習Python的學員們，今天分享的是如何判斷jpeg圖片完整性示例。
用擴展名判斷文件格式非常簡單，但是有可能是錯誤的。jpeg文件有固定的文件頭，其文件頭的格式如下：
Start Marker | JFIF Marker | Header Length | Identifier
0xff, 0xd8? | 0xff, 0xe0 |? 2-bytes? | "JFIF\0"
所以可以通過文件頭的方式快速判斷文件格式：
def is_jpg(filename):
? data = open(filename,'rb').read(11)
? if data[:4] != '\xff\xd8\xff\xe0' and data[:4]!='\xff\xd8\xff\xe1':
? ? return False
? if data[6:] != 'JFIF\0' and data[6:] != 'Exif\0':
? ? return False
? return True
也可以通過PIL類庫來做判斷：
from PIL import Image
def is_jpg(filename):
? try:
? ? i=Image.open(filename)
? ? return i.format =='JPEG'
? except IOError:
? ? return Fals
應用場景：判斷image文件夾中的jpeg文件是否完整，代碼如下：
#coding=utf-8
#summary: 判斷圖片的有效性
import io
import os
from PIL import Image
#判斷文件是否為有效（完整）的圖片
#輸入參數為文件路徑
#會出現漏檢的情況
def IsValidImage(pathfile):
bValid = True
try:
? Image.open(pathfile).verify()
except:
? bValid = False
return bValid
def is_valid_jpg(jpg_file):
? """判斷JPG文件下載是否完整
? """
? if jpg_file.split('.')[-1].lower() == 'jpg':
? ? with open(jpg_file, 'rb') as f:
? ? ? f.seek(-2, 2)
? ? ? return f.read() == '\xff\xd9' #判定jpg是否包含結束字段
? else:
? ? return True
#利用PIL庫進行jpeg格式判定，但有些沒有結束字段的文件檢測不出來
def is_jpg(filename):
? try:
? ? i=Image.open(filename)
? ? return i.format =='JPEG'
? except IOError:
? ? return False
allfiles=os.listdir('image')
log_file=open('img_lossinfo.txt','w')
log = open('img_r.txt','w')
log_w=open('img_w.txt','w')
log1=open('img_jpeg.txt','w')
log2=open('img_notjpg.txt','w')
for i in allfiles:
#if 1:
? ? if i[-4:]=='.jpg':
? ? ? ? f=os.path.join('image',i)
? ? ? ? value=IsValidImage(f)
? ? ? ? if not value:
? ? ? ? ? ? log_file.write(i+'\n')
? ? ? ? if is_valid_jpg(f):
? ? ? ? ? ? print f
? ? ? ? ? ? log.write(i+'\n')
? ? ? ? else:
? ? ? ? ? ? log_w.write(i+'\n')
? ? ? ? if is_jpg(f):
? ? ? ? ? ? log1.write(i+'\n')
? ? ? ? else:
? ? ? ? ? ? log2.write(i+'\n')
以上就是關于小猿圈Python開發如何判斷jpeg圖片完整性示例的全部內容，希望能給大家一個參考，最后想要了解更多關于Python和人工智能方面內容的小伙伴，可以關注小猿圈每天的動態，會不定期的為大家更新知識。
