PyMuPDF转图片

最新推荐文章于 2024-02-01 09:18:40 发布

QuickservedNoodle

最新推荐文章于 2024-02-01 09:18:40 发布

阅读量258

点赞数

CC 4.0 BY-SA版权

分类专栏：笔记文章标签： python

本文链接：https://blog.youkuaiyun.com/QuickservedNoodle/article/details/129262739

笔记专栏收录该内容

4 篇文章

订阅专栏

文章讲述了由于API更新，一些在线教程中的PDF转图片代码出现错误，作者提供了修正后的代码示例，包括fitz.Matrix方法的使用和保存图片的调整。代码示例使用Python的fitz库进行PDF到图片的转换。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

应该是因为API更新，所以网上很多的教程都会报错。这里记录一下。

修正内容：

fitz.Matrix(zoom_x, zoom_y).preRotate(rotate)-->fitz.Matrix(zoom_x, zoom_y).prerotate(rotate)

page.getPixmap(matrix=trans, alpha=False)-->page.get_pixmap(matrix=trans, alpha=False)

pm.writePNG(pic_pwd)-->pm.save(pic_pwd)

PDF转图片的功能

import fitz
import os

# pdf路径
tmp = r"."

export_file = r"./output"

save_path = r"./output"
os.makedirs(export_file, exist_ok=True)
pdf_dir = [i for i in os.listdir(tmp) if os.path.splitext(i)[-1] == ".pdf"]

def pdf_to_jpg(name):
    # 拼接pdf的文件路径
    pwd_name = os.path.join(tmp, name)
    with fitz.open(pwd_name) as doc:
        # 将文件名同我们的保存路径拼接起来（保存图片的文件夹）
        dir_name = os.path.splitext(name)[0]
        pdf_name = os.path.join(export_file, dir_name)
        # print(pdf_name)
        temp = 0
        # （保存图片的文件夹）不存咋则生成
        # exsitsdir.judge(pdf_name)
        os.makedirs(pdf_name, exist_ok=True)
        for pg in range(doc.page_count):
            page = doc[pg]
            temp += 1
            rotate = int(0)
            # 每个尺寸的缩放系数为2，这将为我们生成分辨率提高四倍的图像。
            zoom_x = 2.0
            zoom_y = 2.0
            trans = fitz.Matrix(zoom_x, zoom_y).prerotate(rotate)
            pm = page.get_pixmap(matrix=trans, alpha=False)

            pic_name = '{}.jpg'.format(temp)
            # 拼接生成pdf的文件路径
            pic_pwd = os.path.join(pdf_name, pic_name)
            # print(pic_pwd)
            pm.save(pic_pwd)

def main():
    #多线程
    # pool = Pool(2)
    # for i in pdf_dir:
    #     res = pool.apply_async(pdf_to_jpg, (i,))
    # pool.close()
    # pool.join()
    #单线程
    for i in pdf_dir:
        pdf_to_jpg(i)

if __name__ == '__main__':
    main()