最近需要把扫描的PDF转换成文本,试用了pdfminer,pypdf2等工具,解析图片的效果都不太好,用起来也比较麻烦,后来试用了PyMuPDF,相对其它工具,它最新版本屏蔽了更多细节,围绕Page进行操作,调用非常方便。除了PDF它还支持解析epub等电子书格式。
目前网上例程大多是只抽取图片,没有同时转换图片和文字,且因为旧版本API,大多已无法正常运行。本文将示例其用法。
$ pip install pymupdf==1.18.19
如果提取找不到fiz.h,建议更新pip版本
$ pip3.6 install --upgrade pip
def parse(inpath, outpath): remove(TMPDIR) # 清除临时目录 os.mkdir(TMPDIR) remove(outpath) # 清除输出文件 t0 = time.clock() doc = fitz.open(inpath) lenXREF = doc.xrefLength() print("文件名:{}, 页数: {}, 对象: {}".format(inpath, len(doc), lenXREF - 1)) imgcount = 0 for i,page in enumerate(doc): t1 = time.clock() # 文字 text = page.get_text() if len(text) > 0: write_file(outpath, text, 'a') # 图片 imglist = page.get_images() # 解析该页中图片 for item in imglist: xref = item[0] pix = fitz.Pixmap(doc, xref) new_name = "{}.png".format(imgcount) # 如果pix.n<5,可以直接存为PNG path = os.path.join(TMPDIR, new_name) if pix.n < 5: pix.writePNG(path) # 否则先转换CMYK else: pix0 = fitz.Pixmap(fitz.csRGB, pix) pix0.writePNG(path) pix0 = None pix = None if OCR_ONLINE: text = img_to_str_baidu(path) else: text = img_to_str_tesseract(path) print("img->text", text) write_file(outpath, text, 'a') write_file(outpath, '\n' + '-----------' + '\n', 'a') imgcount += 1 print("page {} 运行时间:{}s".format(i, {t1 - t0}))
完整例程请见参考部分
帮助文档 https://pymupdf.readthedocs.io/en/latest/tutorial.html
源码地址 https://github.com/pymupdf/PyMuPDF
完整例程 https://github.com/xieyan0811/pdfconv.git