python转换并提取pdf文件中的图片

复制代码
#安装fitz包
pip install pymupdf 

脚本如下所示:

复制代码
import fitz
import re
import os
import time
import sys

arguments = sys.argv

for arg in arguments:
    print(arg)


def file_name_list(base_dir):
    for i, j, k in os.walk(base_dir):
        name = [i.replace('.pdf', '') for i in k]
        return name


def pdfExtractPic(filePath, pic_path):
    ret = {}
    try:
        iNum = 0
        with fitz.open(filePath) as doc:
            for page in range(len(doc)):
                for image in doc.get_page_images(page):
                    xref = image[0]
                    pix = fitz.Pixmap(doc, xref)
                    iNum += 1
                    fileName = ("%s\%d.png"%(pic_path, iNum))
                    if pix.n < 5:
                        pix.save(fileName)
                    else:
                        pix1 = fitz.Pixmap(fitz.csRGB, pix)
                        pix1.save(fileName)
                        pix1 = None

    except Exception as e:
        print(e)

    return ret

 
def pdf2image(path, pic_path, image_name):
    try:
        with fitz.open(path) as pdf:
            for pg in range(0, pdf.page_count):
                page = pdf[pg]
        # 设置缩放和旋转系数,zoom_x, zoom_y取相同值,表示等比例缩放
                mat = fitz.Matrix(2, 2)
                pm = page.get_pixmap(matrix=mat, alpha=False)
        # if width or height > 2000 pixels, don't enlarge the image
                if pm.width > 2000 or pm.height > 2000:
                    pm = page.get_pixmap(matrix=fitz.Matrix(1, 1), alpha=False)
                page_num = pg + 1  # 页码从1开始
                new_name = '%s_img_%d.png' % (image_name, page_num)
                pm.save(os.path.join(pic_path, new_name)) 
    except Exception as error:
        print(error)                
 
 
if __name__ == '__main__':
    if len(arguments) < 2:
        print("请输入pdf所在目录的路径.")
        sys.exit(1)
        
    base_dir = arguments[1]

    name_list = file_name_list(base_dir)
    start = time.time()
    for name in name_list:
        image_name = name
        file_name = r'\%s.pdf' % name  # PDF 文件路径
        file_path = base_dir + file_name
        pic_name = r'\pic\%s' % name  # 存放图片的文件夹
        pic_path = base_dir + pic_name
        try:
            os.makedirs(pic_path)   # 根据路径,创建对应路径下的文件夹
        except Exception as error:
            print(error)        
        pdf2image(file_path, pic_path, image_name)
        #print(pic_path)
        pic_path += "\\sub\\"
        try:
            os.makedirs(pic_path)   # 根据路径,创建对应路径下的文件夹
        except Exception as error:
            print(error)   
        pdfExtractPic(file_path, pic_path)
    end = time.time()
    print('task is over: %.2f' % (end-start))

如果执行的时候报错:

复制代码
RuntimeError: Directory 'static' does not exist

说明fitz安装错了版本,参考:PyMuPDF: AttributeError:模块'fitz'没有属性'open'_attributeerror: module 'fitz' has no attribute 'op-CSDN博客

相关推荐
1024小神13 分钟前
tauri项目在windows上的c盘没有权限写入文件
c语言·开发语言·windows
老虎062721 分钟前
数据结构(Java)--位运算
java·开发语言·数据结构
yanjiaweiya22 分钟前
云原生-集群管理续
java·开发语言·云原生
Swift社区23 分钟前
Swift 解 LeetCode 320:一行单词有多少种缩写可能?用回溯找全解
开发语言·leetcode·swift
写不出来就跑路28 分钟前
暑期实习感悟与经验分享:从校园到职场的成长之路
java·开发语言·经验分享·spring boot
灵智工坊LingzhiAI2 小时前
人体坐姿检测系统项目教程(YOLO11+PyTorch+可视化)
人工智能·pytorch·python
weixin_472339466 小时前
高效处理大体积Excel文件的Java技术方案解析
java·开发语言·excel
枯萎穿心攻击7 小时前
响应式编程入门教程第二节:构建 ObservableProperty<T> — 封装 ReactiveProperty 的高级用法
开发语言·unity·c#·游戏引擎
Eiceblue8 小时前
【免费.NET方案】CSV到PDF与DataTable的快速转换
开发语言·pdf·c#·.net
m0_555762909 小时前
Matlab 频谱分析 (Spectral Analysis)
开发语言·matlab