0

语境：

我有正在使用的 PDF 文件。我正在使用ocr从这些文档中提取文本并且能够做到这一点，我必须将我的 pdf 文件转换为图像。我目前使用模块的convert_from_path功能，pdf2image但时间效率非常低（9page pdf 需要 9 分钟）。

问题：

我正在寻找一种方法来加速此过程或另一种将我的 PDF 文件转换为图像的方法。

附加信息：

我知道thread_count函数中有一个参数，但经过几次尝试，它似乎没有任何区别。

这是我正在使用的整个功能：

def pdftoimg(fic,output_folder):
# Store all the pages of the PDF in a variable 
pages = convert_from_path(fic, dpi=500,output_folder=output_folder,thread_count=9, poppler_path=r'C:\Users\Vincent\Documents\PDF\poppler-21.02.0\Library\bin') 

image_counter = 0

# Iterate through all the pages stored above 
for page in pages: 
    filename = "page_"+str(image_counter)+".jpg"
    page.save(output_folder+filename, 'JPEG') 
    image_counter = image_counter + 1
    
for i in os.listdir(output_folder):
    if i.endswith('.ppm'):
        os.remove(output_folder+i)

链接到 convert_from_path 参考。

4

1 回答 1

0

我使用另一个名为的模块找到了该问题的答案，该模块fitz是一个 python 绑定到MuPDF.

首先安装 PyMuPDF：

可以在此处找到文档，但对于 Windows 用户来说，它相当简单：

pip install PyMuPDF

然后导入 fitz 模块：

import fitz
print(fitz.__doc__)

>>>PyMuPDF 1.18.13: Python bindings for the MuPDF 1.18.0 library.
>>>Version date: 2021-05-05 06:32:22.
>>>Built for Python 3.7 on win32 (64-bit).

打开您的文件并将每一页保存为图像：

get_pixmap() 方法接受不同的参数，允许您控制图像（变化、分辨率、颜色...），因此我建议您在此处更改文档。

def convert_pdf_to_image(fic):
    #open your file
    doc = fitz.open(fic)
    #iterate through the pages of the document and create a RGB image of the page
    for page in doc:
        pix = page.get_pixmap()
        pix.save("page-%i.png" % page.number)

希望这对其他人有帮助。

于 2021-05-20T09:48:43.143 回答