
使用pymupdf(fitz)提取pdf图像时,常因忽略pixmap坐标系差异、未处理alpha通道及颜色空间转换,导致图像倒置、镜像或色彩失真;本文提供完整解决方案,涵盖pixmap构建、垂直翻转、rgb校准与内存释放。
PDF中的图像数据并非以常规光栅图像形式直接存储,而是通过PDF内容流+资源字典+XObject引用组合描述,其坐标系原点位于左下角(与opencv/PIL的左上角原点相反),且原始Pixmap对象默认按PDF内部格式(如CMYK、索引色或带Alpha的灰度)编码。直接调用 document.extract_image() 返回的字节流可能已丢失方向元信息或经历隐式解码,造成视觉上的“倒置”和“颜色异常”——这正是您代码中图像上下颠倒、色彩偏黄/发灰的根本原因。
✅ 正确做法是:绕过 extract_image(),直接基于XRef构造 fitz.Pixmap 对象,再手动转换为numpy数组并做坐标系对齐:
import fitz import numpy as np import cv2 import os 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 try: # ✅ Step 1: Create Pixmap directly from XRef (preserves native orientation & color info) pix = fitz.Pixmap(document, xref) # ✅ Step 2: Handle color space — convert to RGB if needed (critical for CMYK/Gray+Alpha) if pix.n > 4: # e.g., CMYK + Alpha → drop Alpha, convert to RGB pix = fitz.Pixmap(fitz.csRGB, pix) elif pix.n == 4: # RGB + Alpha → remove alpha channel pix = fitz.Pixmap(fitz.csRGB, pix) elif pix.n == 1: # Grayscale → expand to RGB pix = fitz.Pixmap(fitz.csRGB, pix) # ✅ Step 3: Convert to NumPy array (H×W×N), then FLIP VERTICALLY to match OpenCV/PIL origin img_np = np.frombuffer(pix.samples, dtype=np.uint8).reshape(pix.h, pix.w, pix.n) img_np_flipped = cv2.flip(img_np, 0) # ← This fixes "upside-down" # ✅ Step 4: Convert BGR→RGB before saving (cv2.imwrite uses BGR by default) img_rgb = cv2.cvtColor(img_np_flipped, cv2.COLOR_BGR2RGB) # ✅ Step 5: Save with PIL or cv2 (PIL handles RGB natively) output_path = os.path.join(page_folder, f"image_{img_index}.png") Image.fromarray(img_rgb).save(output_path) print(f"✓ Saved: {output_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 # ? Essential: release Pixmap memory explicitly document.close()
? 关键要点说明:
- Pixmap ≠ Raw Bytes:extract_image() 返回的是PDF解码后的JPEG/PNG字节流,可能已被压缩或重采样;而 fitz.Pixmap(document, xref) 直接从PDF对象重建像素缓冲区,保留原始分辨率与方向语义。
- 垂直翻转不可省略:PDF的 (0,0) 在左下,Pixmap的 .samples 数组按PDF坐标顺序填充(即第0行=页面最底行),因此 cv2.flip(…, 0) 是校正显示方向的必要步骤。
- 颜色空间必须显式归一化:PDF支持CMYK、DeviceN、Indexed等非RGB色彩模型;不转换会导致OpenCV误读通道(如将CMYK当BGR),引发严重色偏。pix = fitz.Pixmap(fitz.csRGB, pix) 强制转换为标准sRGB。
- 及时释放Pixmap:pix = None 不仅是良好实践,更是避免内存泄漏(尤其处理大PDF或多页时)的强制要求。
? 额外建议:
- 若仍存在轻微色差,可尝试在保存前添加Gamma校正(img_rgb = np.clip(img_rgb ** 1.1 * 255, 0, 255).astype(np.uint8));
- 对含透明背景的图像,如需白底导出,可在转换前用 pix = fitz.Pixmap(fitz.csRGB, pix, 255) 指定白色背景;
- 避免混用 cv2.imwrite(BGR)与 PIL.Image.fromarray(RGB)——统一用PIL更安全,因其自动识别通道数。
遵循上述流程后,提取图像将与adobe Illustrator手动导出效果完全一致:方向正确、色彩准确、无畸变。