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

? 关键注意事项

VisualizeAI
VisualizeAI

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

相关专题

更多
java数据库连接教程大全
java数据库连接教程大全

本专题整合了java数据库连接相关教程,阅读专题下面的文章了解更多详细内容。

20

2026.01.15

Java音频处理教程汇总
Java音频处理教程汇总

本专题整合了java音频处理教程大全,阅读专题下面的文章了解更多详细内容。

5

2026.01.15

windows查看wifi密码教程大全
windows查看wifi密码教程大全

本专题整合了windows查看wifi密码教程大全,阅读专题下面的文章了解更多详细内容。

26

2026.01.15

浏览器缓存清理方法汇总
浏览器缓存清理方法汇总

本专题整合了浏览器缓存清理教程汇总,阅读专题下面的文章了解更多详细内容。

2

2026.01.15

ps图片相关教程汇总
ps图片相关教程汇总

本专题整合了ps图片设置相关教程合集,阅读专题下面的文章了解更多详细内容。

7

2026.01.15

ppt一键生成相关合集
ppt一键生成相关合集

本专题整合了ppt一键生成相关教程汇总,阅读专题下面的的文章了解更多详细内容。

3

2026.01.15

php图片上传教程汇总
php图片上传教程汇总

本专题整合了php图片上传相关教程,阅读专题下面的文章了解更多详细教程。

2

2026.01.15

phpstorm相关教程大全
phpstorm相关教程大全

本专题整合了phpstorm相关教程汇总,阅读专题下面的文章了解更多详细内容。

4

2026.01.15

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

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

16

2026.01.15

热门下载

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

精品课程

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

共4课时 | 0.8万人学习

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号