第一節:python提取PDF文檔中的圖片


由於項目需要將PDF文檔當中的圖片轉換成圖片,所以參考了這篇文章https://blog.csdn.net/qq_15969343/article/details/81673302后項目得以解決。

1、安裝第三方類庫pymupdf:pip install pymupdf

2、安裝完成后直接上代碼,代碼如下:

import fitz
import time
import re
import os
def pdf2pic(path, pic_path):
    t0 = time.clock()                          # 生成圖片初始時間
    checkXO = r"/Type(?= */XObject)"           # 使用正則表達式來查找圖片
    checkIM = r"/Subtype(?= */Image)"
    doc = fitz.open(path)                      # 打開pdf文件
    imgcount = 0                               # 圖片計數
    lenXREF = doc._getXrefLength()             # 獲取對象數量長度
    
    # 打印PDF的信息
    print("文件名:{}, 頁數: {}, 對象: {}".format(path, len(doc), lenXREF - 1))
    
    # 遍歷每一個對象
    for i in range(1, lenXREF):
        text = doc._getXrefString(i)            # 定義對象字符串
        isXObject = re.search(checkXO, text)    # 使用正則表達式查看是否是對象
        isImage = re.search(checkIM, text)      # 使用正則表達式查看是否是圖片
        if not isXObject or not isImage:        # 如果不是對象也不是圖片,則continue
            continue
        imgcount += 1
        pix = fitz.Pixmap(doc, i)               # 生成圖像對象
        new_name = "圖片{}.png".format(imgcount) # 生成圖片的名稱
        if pix.n < 5:                           # 如果pix.n<5,可以直接存為PNG
            pix.writePNG(os.path.join(pic_path, new_name))
        else:                                   # 否則先轉換CMYK
            pix0 = fitz.Pixmap(fitz.csRGB, pix)
            pix0.writePNG(os.path.join(pic_path, new_name))
            pix0 = None
        pix = None                              # 釋放資源
        t1 = time.clock()                       # 圖片完成時間
        print("運行時間:{}s".format(t1 - t0))
        print("提取了{}張圖片".format(imgcount))
        
if __name__=='__main__':
    path = r"C:\Users\lenovo\Desktop\數據.pdf"
    pic_path = r'C:\Users\lenovo\Desktop\圖片'
    # 創建保存圖片的文件夾
    if os.path.exists(pic_path):
        print("文件夾已存在,不必重新創建!")
        pass
    else:
        os.mkdir(pic_path)
    pdf2pic(path, pic_path)

3、運行結果

文件名:C:\Users\lenovo\Desktop\數據.pdf, 頁數: 51, 對象: 156
運行時間:1.0037559488187855s
提取了1張圖片
運行時間:1.9240614402553362s
提取了2張圖片
運行時間:2.8580821293209087s
提取了3張圖片
運行時間:3.59311390384999s
提取了4張圖片
運行時間:4.055301359322903s
提取了5張圖片
運行時間:4.861761705280556s


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM