小猿圈Python開發如何判斷jpeg圖片完整性示例

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和人工智能方面內容的小夥伴，能夠關注小猿圈天天的動態，會不按期的爲你們更新知識。