0

0

如何正确提取PDF中的图像:解决PyMuPDF导出图片旋转、倒置与颜色异常问题

霞舞

霞舞

发布时间:2026-01-15 17:15:09

|

404人浏览过

|

来源于php中文网

原创

如何正确提取PDF中的图像:解决PyMuPDF导出图片旋转、倒置与颜色异常问题

使用pymupdf(fitz)提取pdf图像时,常出现图像上下颠倒、左右镜像或色彩失真等问题,根本原因在于pdf坐标系(原点在左下角)与opencv/pil(原点在左上角)不一致,且pixmap的alpha通道和色彩空间未正确处理。本文提供完整、健壮的解决方案。

PDF内部采用数学笛卡尔坐标系:页面原点位于左下角,Y轴向上为正;而OpenCV、NumPy数组及大多数图像库(包括PIL)约定图像原点在左上角,Y轴向下为正。因此,直接将PyMuPDF的Pixmap.samples按(h, w, n)重塑后直接使用,会导致图像垂直翻转(即视觉上“倒置”)。此外,document.extract_image()返回的是原始编码字节(如JPEG/JBIG2/Flate),可能丢失变换信息(如旋转矩阵、色彩校准),且不处理PDF中常见的CMYK、灰度带Alpha等非标准色彩空间——这正是你手动在Illustrator中无问题、但代码提取却变色的根本原因。

✅ 正确做法是:绕过extract_image(),直接基于xref构造Pixmap对象,并显式处理色彩空间与坐标翻转:

import fitz
import numpy as np
import cv2
import os

pdf_path = '/content/drive/MyDrive/Wettbewerb Aktuell/1803_AusgGesa-pages-2.pdf'
output_directory = '/content/drive/MyDrive/Wettbewerb Aktuell/images/'

document = fitz.open(pdf_path)
start_page, end_page = 0, min(1, document.page_count - 1)
os.makedirs(output_directory, exist_ok=True)

for page_number in range(start_page, end_page + 1):
    page = document[page_number]
    images = page.get_images(full=True)
    page_folder = os.path.join(output_directory, f"page_{page_number}/")
    os.makedirs(page_folder, exist_ok=True)

    for img_index, img in enumerate(images):
        xref = img[0]  # 图像XREF引用号
        try:
            # ✅ 关键:用xref创建Pixmap(保留PDF原生渲染信息)
            pix = fitz.Pixmap(document, xref)

            # ✅ 处理多通道/Alpha:强制转为RGB(去除Alpha或合并)
            if pix.n - pix.alpha > 3:  # 非标准色彩空间(如CMYK)
                pix = fitz.Pixmap(fitz.csRGB, pix)
            elif pix.alpha:  # 含Alpha通道 → 合并到白色背景
                pix = fitz.Pixmap(fitz.csRGB, pix)
                # 注意:pix.pil_save() 自动处理Alpha,但numpy需手动
            else:
                # 确保为RGB(pix.n == 3)或灰度(pix.n == 1)
                pass

            # ✅ 正确转换为NumPy数组(注意:samples是C-contiguous bytes)
            img_np = np.frombuffer(pix.samples, dtype=np.uint8).reshape(pix.h, pix.w, pix.n)

            # ✅ 垂直翻转:适配PDF(左下原点)→ OpenCV(左上原点)
            img_np = cv2.flip(img_np, 0)

            # ✅ 颜色空间校正:Pixmap默认BGR-like顺序?实际为RGB(但cv2.imwrite要求BGR)
            # 若pix.n == 3,且为RGB,则需转BGR;若已为BGR则跳过
            # PyMuPDF Pixmap.samples 是RGB顺序(文档明确说明),故需转BGR供cv2.imwrite
            if pix.n == 3:
                img_np = cv2.cvtColor(img_np, cv2.COLOR_RGB2BGR)

            # 保存
            img_path = os.path.join(page_folder, f"image_{img_index}.jpg")
            cv2.imwrite(img_path, img_np)
            print(f"✓ Saved: {img_path}")

        except Exception as e:
            print(f"✗ Failed to process image {img_index} on page {page_number}: {e}")
        finally:
            if 'pix' in locals():
                pix = None  # 显式释放Pixmap内存(重要!)

document.close()

? 关键注意事项

造梦阁AI
造梦阁AI

AI小说推文一键成片,你的故事值得被看见

下载
  • 永远优先使用 Pixmap(document, xref) 而非 extract_image():前者执行PDF解释器级渲染,保留CTM(当前变换矩阵)、色彩配置、软掩模等元信息;后者仅解码原始流,易丢失方向与色彩上下文。
  • cv2.flip(img_np, 0) 不可省略:这是修复“倒置”的核心步骤,对应PDF坐标系到图像坐标系的映射。
  • 色彩空间必须显式归一化:PDF支持CMYK、Lab、Indexed等,fitz.Pixmap(fitz.csRGB, pix) 强制转换为sRGB,避免OpenCV误读通道。
  • 内存管理:Pixmap对象占用较大内存,务必在循环末尾设为None或使用with上下文(PyMuPDF v1.23+支持)。
  • 调试建议:若仍有色偏,尝试用 pix.save("debug.png") 直接保存Pixmap(它自动处理翻转与色彩),对比是否与cv2.imwrite结果一致——若一致则问题在后续处理;若不一致,说明np.frombuffer重塑有误(检查pix.h/pix.w/pix.n是否匹配)。

通过以上方法,你将获得与Illustrator手动导出完全一致的方向、比例与色彩效果。

相关专题

更多
Golang gRPC 服务开发与Protobuf实战
Golang gRPC 服务开发与Protobuf实战

本专题系统讲解 Golang 在 gRPC 服务开发中的完整实践,涵盖 Protobuf 定义与代码生成、gRPC 服务端与客户端实现、流式 RPC(Unary/Server/Client/Bidirectional)、错误处理、拦截器、中间件以及与 HTTP/REST 的对接方案。通过实际案例,帮助学习者掌握 使用 Go 构建高性能、强类型、可扩展的 RPC 服务体系,适用于微服务与内部系统通信场景。

6

2026.01.15

公务员递补名单公布时间 公务员递补要求
公务员递补名单公布时间 公务员递补要求

公务员递补名单公布时间不固定,通常在面试前,由招录单位(如国家知识产权局、海关等)发布,依据是原入围考生放弃资格,会按笔试成绩从高到低递补,递补考生需按公告要求限时确认并提交材料,及时参加面试/体检等后续环节。要求核心是按招录单位公告及时响应、提交材料(确认书、资格复审材料)并准时参加面试。

37

2026.01.15

公务员调剂条件 2026调剂公告时间
公务员调剂条件 2026调剂公告时间

(一)符合拟调剂职位所要求的资格条件。 (二)公共科目笔试成绩同时达到拟调剂职位和原报考职位的合格分数线,且考试类别相同。 拟调剂职位设置了专业科目笔试条件的,专业科目笔试成绩还须同时达到合格分数线,且考试类别相同。 (三)未进入原报考职位面试人员名单。

51

2026.01.15

国考成绩查询入口 国考分数公布时间2026
国考成绩查询入口 国考分数公布时间2026

笔试成绩查询入口已开通,考生可登录国家公务员局中央机关及其直属机构2026年度考试录用公务员专题网站http://bm.scs.gov.cn/pp/gkweb/core/web/ui/business/examResult/written_result.html,查询笔试成绩和合格分数线,点击“笔试成绩查询”按钮,凭借身份证及准考证进行查询。

8

2026.01.15

Java 桌面应用开发(JavaFX 实战)
Java 桌面应用开发(JavaFX 实战)

本专题系统讲解 Java 在桌面应用开发领域的实战应用,重点围绕 JavaFX 框架,涵盖界面布局、控件使用、事件处理、FXML、样式美化(CSS)、多线程与UI响应优化,以及桌面应用的打包与发布。通过完整示例项目,帮助学习者掌握 使用 Java 构建现代化、跨平台桌面应用程序的核心能力。

65

2026.01.14

php与html混编教程大全
php与html混编教程大全

本专题整合了php和html混编相关教程,阅读专题下面的文章了解更多详细内容。

36

2026.01.13

PHP 高性能
PHP 高性能

本专题整合了PHP高性能相关教程大全,阅读专题下面的文章了解更多详细内容。

75

2026.01.13

MySQL数据库报错常见问题及解决方法大全
MySQL数据库报错常见问题及解决方法大全

本专题整合了MySQL数据库报错常见问题及解决方法,阅读专题下面的文章了解更多详细内容。

20

2026.01.13

PHP 文件上传
PHP 文件上传

本专题整合了PHP实现文件上传相关教程,阅读专题下面的文章了解更多详细内容。

33

2026.01.13

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 0.7万人学习

Rust 教程
Rust 教程

共28课时 | 4.4万人学习

Git 教程
Git 教程

共21课时 | 2.7万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号