
使用pymupdf(fitz)从pdf中提取图像时,常因忽略pdf坐标系差异、pixmap色彩空间处理及图像翻转逻辑,导致提取结果倒置、旋转或颜色失真;本文提供完整解决方案,涵盖pixmap创建、色彩空间转换、垂直翻转与rgb通道校正。
PDF采用与屏幕相反的坐标系:原点位于左下角,Y轴向上为正;而OpenCV/PIL等图像库默认原点在左上角,Y轴向下为正。这直接导致原始Pixmap数据按行存储时,首行为PDF页面底部像素——若直接转为NumPy数组并保存,图像将呈现上下颠倒(inverted)状态。此外,Pixmap的pix.samples是未经解码的原始字节流,其通道顺序、色彩空间(如CMYK、灰度、带Alpha的RGB)均可能与预期不符,引发颜色偏移(如发绿、泛红)或透明通道干扰。
以下为修正后的完整提取流程,已整合关键修复点:
import os
import numpy as np
import cv2
import fitz # PyMuPDF
pdf_path = '/content/drive/MyDrive/Wettbewerb Aktuell/1803_AusgGesa-pages-2.pdf'
document = fitz.open(pdf_path)
start_page = 0
end_page = min(1, document.page_count - 1)
output_directory = '/content/drive/MyDrive/Wettbewerb Aktuell/images/'
os.makedirs(output_directory, exist_ok=True)
for page_number in range(start_page, end_page + 1):
page_folder = os.path.join(output_directory, f"page_{page_number}/")
os.makedirs(page_folder, exist_ok=True)
page = document[page_number]
images = page.get_images(full=True)
for img_index, img in enumerate(images):
xref = img[0] # XREF of the image object
# ✅ 步骤1:显式创建Pixmap(关键!避免extract_image的隐式解码缺陷)
pix = fitz.Pixmap(document, xref)
# ✅ 步骤2:统一转换为sRGB色彩空间(解决CMYK/灰度/多通道颜色异常)
if pix.n > 4: # 超过RGBA(4通道),强制转RGB
pix = fitz.Pixmap(fitz.csRGB, pix)
elif pix.n == 4: # 带Alpha的RGBA → RGB(丢弃Alpha,避免混合失真)
pix = fitz.Pixmap(fitz.csRGB, pix)
elif pix.n == 1: # 灰度图 → RGB(避免cv2.cvtColor报错)
pix = fitz.Pixmap(fitz.csRGB, pix)
# ✅ 步骤3:转换为NumPy数组并垂直翻转(修正PDF坐标系颠倒)
img_np = np.frombuffer(pix.samples, dtype=np.uint8).reshape(pix.h, pix.w, pix.n)
img_np_flipped = cv2.flip(img_np, 0) # 沿X轴翻转 → 恢复正向显示
# ✅ 步骤4:通道校正(Pixmap为BGR顺序?不!PyMuPDF samples是RGB顺序,但cv2.imwrite默认BGR)
# 因此需转BGR保存,或用PIL保存(更稳妥)
# 方案A(推荐):用cv2.imwrite + BGR转换
cv2.imwrite(os.path.join(page_folder, f"image_{img_index}.jpg"),
cv2.cvtColor(img_np_flipped, cv2.COLOR_RGB2BGR))
# 方案B(备选):用PIL保存(自动处理RGB)
# Image.fromarray(img_np_flipped).save(os.path.join(page_folder, f"image_{img_index}.png"))
print(f"✅ Saved: {os.path.join(page_folder, f'image_{img_index}.jpg')}")
pix = None # ⚠️ 必须释放Pixmap内存,防止OOM
document.close()关键注意事项:
- 勿用 document.extract_image():该方法返回的是PDF内嵌的原始编码数据(如JPEG字节流),未经过PyMuPDF的渲染层处理,无法自动适配坐标系和色彩空间,易出错。
- 必须创建 Pixmap:fitz.Pixmap(document, xref) 才能获得经PyMuPDF渲染引擎处理后的像素数据,支持色彩空间转换与尺寸信息访问。
- cv2.flip(img_np, 0) 不可省略:这是修正倒置的核心操作,对应PDF坐标系与图像库坐标的本质差异。
- 色彩空间转换优先级高于翻转:先确保pix为标准RGB格式(pix.n == 3),再执行reshape与flip,否则reshape维度会错误。
- 及时释放Pixmap:循环内pix = None可触发__del__释放底层内存,处理多页PDF时至关重要。
若仍遇颜色异常(如偏青、发灰),可检查PDF源文件是否使用了非标准色彩配置文件(如Adobe RGB),此时建议在Pixmap创建后添加pix.set_dpi(300)提升采样精度,或导出为PNG(保留Alpha)再用专业工具校色。最终效果将与Illustrator手动导出一致——方向正确、色彩真实、细节完整。










