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对象,并显式处理色彩空间与坐标翻转:

What-the-Diff
What-the-Diff

检查请求差异,自动生成更改描述

下载
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()

? 关键注意事项

  • 永远优先使用 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手动导出完全一致的方向、比例与色彩效果。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
C# ASP.NET Core微服务架构与API网关实践
C# ASP.NET Core微服务架构与API网关实践

本专题围绕 C# 在现代后端架构中的微服务实践展开,系统讲解基于 ASP.NET Core 构建可扩展服务体系的核心方法。内容涵盖服务拆分策略、RESTful API 设计、服务间通信、API 网关统一入口管理以及服务治理机制。通过真实项目案例,帮助开发者掌握构建高可用微服务系统的关键技术,提高系统的可扩展性与维护效率。

16

2026.03.11

Go高并发任务调度与Goroutine池化实践
Go高并发任务调度与Goroutine池化实践

本专题围绕 Go 语言在高并发任务处理场景中的实践展开,系统讲解 Goroutine 调度模型、Channel 通信机制以及并发控制策略。内容包括任务队列设计、Goroutine 池化管理、资源限制控制以及并发任务的性能优化方法。通过实际案例演示,帮助开发者构建稳定高效的 Go 并发任务处理系统,提高系统在高负载环境下的处理能力与稳定性。

23

2026.03.10

Kotlin Android模块化架构与组件化开发实践
Kotlin Android模块化架构与组件化开发实践

本专题围绕 Kotlin 在 Android 应用开发中的架构实践展开,重点讲解模块化设计与组件化开发的实现思路。内容包括项目模块拆分策略、公共组件封装、依赖管理优化、路由通信机制以及大型项目的工程化管理方法。通过真实项目案例分析,帮助开发者构建结构清晰、易扩展且维护成本低的 Android 应用架构体系,提升团队协作效率与项目迭代速度。

75

2026.03.09

JavaScript浏览器渲染机制与前端性能优化实践
JavaScript浏览器渲染机制与前端性能优化实践

本专题围绕 JavaScript 在浏览器中的执行与渲染机制展开,系统讲解 DOM 构建、CSSOM 解析、重排与重绘原理,以及关键渲染路径优化方法。内容涵盖事件循环机制、异步任务调度、资源加载优化、代码拆分与懒加载等性能优化策略。通过真实前端项目案例,帮助开发者理解浏览器底层工作原理,并掌握提升网页加载速度与交互体验的实用技巧。

95

2026.03.06

Rust内存安全机制与所有权模型深度实践
Rust内存安全机制与所有权模型深度实践

本专题围绕 Rust 语言核心特性展开,深入讲解所有权机制、借用规则、生命周期管理以及智能指针等关键概念。通过系统级开发案例,分析内存安全保障原理与零成本抽象优势,并结合并发场景讲解 Send 与 Sync 特性实现机制。帮助开发者真正理解 Rust 的设计哲学,掌握在高性能与安全性并重场景中的工程实践能力。

218

2026.03.05

PHP高性能API设计与Laravel服务架构实践
PHP高性能API设计与Laravel服务架构实践

本专题围绕 PHP 在现代 Web 后端开发中的高性能实践展开,重点讲解基于 Laravel 框架构建可扩展 API 服务的核心方法。内容涵盖路由与中间件机制、服务容器与依赖注入、接口版本管理、缓存策略设计以及队列异步处理方案。同时结合高并发场景,深入分析性能瓶颈定位与优化思路,帮助开发者构建稳定、高效、易维护的 PHP 后端服务体系。

420

2026.03.04

AI安装教程大全
AI安装教程大全

2026最全AI工具安装教程专题:包含各版本AI绘图、AI视频、智能办公软件的本地化部署手册。全篇零基础友好,附带最新模型下载地址、一键安装脚本及常见报错修复方案。每日更新,收藏这一篇就够了,让AI安装不再报错!

168

2026.03.04

Swift iOS架构设计与MVVM模式实战
Swift iOS架构设计与MVVM模式实战

本专题聚焦 Swift 在 iOS 应用架构设计中的实践,系统讲解 MVVM 模式的核心思想、数据绑定机制、模块拆分策略以及组件化开发方法。内容涵盖网络层封装、状态管理、依赖注入与性能优化技巧。通过完整项目案例,帮助开发者构建结构清晰、可维护性强的 iOS 应用架构体系。

222

2026.03.03

C++高性能网络编程与Reactor模型实践
C++高性能网络编程与Reactor模型实践

本专题围绕 C++ 在高性能网络服务开发中的应用展开,深入讲解 Socket 编程、多路复用机制、Reactor 模型设计原理以及线程池协作策略。内容涵盖 epoll 实现机制、内存管理优化、连接管理策略与高并发场景下的性能调优方法。通过构建高并发网络服务器实战案例,帮助开发者掌握 C++ 在底层系统与网络通信领域的核心技术。

33

2026.03.03

热门下载

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

精品课程

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

共4课时 | 22.5万人学习

Rust 教程
Rust 教程

共28课时 | 6.8万人学习

Git 教程
Git 教程

共21课时 | 4.1万人学习

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

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