之前想了很多種辦法來(lái)解析PDF文件內(nèi)容(Python 讀取PDF文件),一般針對(duì)電子版word或者Excel轉(zhuǎn)化為PDF的PDF文件具有不錯(cuò)的效果,但是依然不能解析圖片(圖片中文字)內(nèi)容,如果PDF是掃描版呢,那種方法就不行了。
所以我又想了另一個(gè)辦法,就是先把PDF文件轉(zhuǎn)化為圖片,然后再將圖片進(jìn)行OCR識(shí)別,得到最終PDF的內(nèi)容。
1 把PDF轉(zhuǎn)化為圖片
可以參考博文:windows下用Python把pdf文件轉(zhuǎn)化為圖片
代碼如下:
import fitz
PDF_path = "你的PDF文件路徑"
imgs_save_path = "圖片保存文件夾路徑"
def PDF_to_imgs(PDF_path, save_path):
# 打開(kāi)PDF文件,生成一個(gè)對(duì)象
doc = fitz.open(PDF_path)
# 將PDF文件的每一頁(yè)都轉(zhuǎn)化為圖片
for pg in range(doc.pageCount):
page = doc[pg]
rotate = int(0)
# 每個(gè)尺寸的縮放系數(shù)為2,這將為我們生成分辨率提高4倍的圖像。
zoom_x = 2
zoom_y = 2
trans = fitz.Matrix(zoom_x, zoom_y).preRotate(rotate)
pm = page.getPixmap(matrix=trans, alpha=False)
pm.writePNG(save_path + '%s.png' % pg)
PDF_to_imgs(PDF_path, imgs_save_path)
2 將圖片進(jìn)行OCR識(shí)別
pytesseract模塊的安裝參考博文:基于python的OCR字符識(shí)別
代碼如下:
import os
import pytesseract
from PIL import Image
imgs_save_path = "C:\\Users\\Administrator\\Desktop\\PDF\\"
txts_save_path = "C:\\Users\\Administrator\\Desktop\\TXT\\"
def img_to_txt(imgs_path, save_path):
# 將文件夾下的所有圖片進(jìn)行OCR識(shí)別
for files, _, file_names in os.walk(imgs_path):
for file_name in file_names:
image = Image.open(files + file_name)
# chi_sim 是中文識(shí)別包,equ 是數(shù)學(xué)公式包,eng 是英文包
content = pytesseract.image_to_string(image, lang="chi_sim")
txt_name = file_name.split(".")[0] + ".txt"
with open(save_path + txt_name, "w") as f:
f.write(content)
img_to_txt(imgs_save_path, txts_save_path)
?
更多文章、技術(shù)交流、商務(wù)合作、聯(lián)系博主
微信掃碼或搜索:z360901061

微信掃一掃加我為好友
QQ號(hào)聯(lián)系: 360901061
您的支持是博主寫(xiě)作最大的動(dòng)力,如果您喜歡我的文章,感覺(jué)我的文章對(duì)您有幫助,請(qǐng)用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧,狠狠點(diǎn)擊下面給點(diǎn)支持吧,站長(zhǎng)非常感激您!手機(jī)微信長(zhǎng)按不能支付解決辦法:請(qǐng)將微信支付二維碼保存到相冊(cè),切換到微信,然后點(diǎn)擊微信右上角掃一掃功能,選擇支付二維碼完成支付。
【本文對(duì)您有幫助就好】元
