使用PyPDF2从PDF中提取的图像是倾斜的，并且其颜色是反转的

Question

我使用以下代码从PDF文件中提取图像：

from PIL import Image
from PyPDF2 import PdfFileReader, generic
from io import BytesIO
from typing import List
import zlib
import sys
import struct
def tiff_header_for_CCITT(width:int, height:int, img_size:int, CCITT_group:int=4) -> bytes:
tiff_header_struct = '<' + '2s' + 'h' + 'l' + 'h' + 'hhll' * 8 + 'l'
return struct.pack(tiff_header_struct,
b'II',
42,
8,
8,
256, 4, 1, width,
257, 4, 1, height,
258, 3, 1, 1,
259, 3, 1, CCITT_group,
262, 3, 1, 0,
273, 4, 1, struct.calcsize(tiff_header_struct),
278, 4, 1, height,
279, 4, 1, img_size,
0
)
def object_to_images(objects:generic.DictionaryObject) -> List:
images:List[Image] = []
for idx, key in enumerate(objects):
obj = objects[key] # object est un mot-clé réservé
if '/Resources' in obj and '/XObject' in obj['/Resources']: # l'objet est un conteneur
images += object_to_images(obj["/Resources"]["/XObject"].getObject()) # récursion
elif obj['/Subtype'] == '/Image':
print(obj)
if "/ImageMask" in obj: # masque d'image, ignoré
continue
img_modes = {'/DeviceRGB': 'RGB',
'/DefaultRGB': 'RGB',
'/DeviceCMYK': 'CMYK',
'/DefaultCMYK': 'CMYK',
'/DeviceGray': 'L',
'/DefaultGray': 'L',
'/Indexed': 'P'}
cspace = obj.get('/ColorSpace').getObject()
indexed = False
if cspace:
if isinstance(cspace, generic.ArrayObject):
if cspace[0] == '/ICCBased':
color_map = cspace[1].getObject()['/N']
if color_map == 1:
colorspace = "P"
elif color_map == 3:
colorspace = "RGB"
elif color_map == 4:
colorspace = "CMYK"
elif cspace[0] == "/Indexed":
c, b, h, l = [v.getObject() for v in cspace]
indexed = True
colorspace = img_modes[c]
else:
colorspace = img_modes[cspace]
width = obj['/Width']
height = obj['/Height']
if "/FlateDecode" in obj["/Filter"]: # Compressé avec zlib
data = zlib.decompress(obj._data)
else:
data = obj._data
if "/DCTDecode" in obj["/Filter"]: # JPEG, rien à faire
img = Image.open(BytesIO(data))
images.append(img)
elif "/JPXDecode" in obj["/Filter"]: # JPEG 2000, rien à faire
img = Image.open(BytesIO(data))
images.append(img)
elif "CCITTFaxDecode" in obj["/Filter"]: # TIFF, rajouter l'en-tête
if obj['/DecodeParms']['/K'] == -1:
CCITT_group = 4
else:
CCITT_group = 3
img_size = len(data)
tiff_header = tiff_header_for_CCITT(width, height, img_size, CCITT_group)
data = tiff_header + data
img = Image.open(BytesIO(data))
images.append(img)
else:
img = Image.frombytes(colorspace, (width, height), data)
if indexed:
img.putpalette(l.getData())
img = img.convert('RGB')
images.append(img)
return images
def pdf_to_images(filename:str) -> List:
images:List[Image] = []
file = PdfFileReader(open(filename, "rb"))
nPages = file.getNumPages()
for i in range(nPages):
page = file.getPage(i)
try:
root_objects = page["/Resources"]["/XObject"].getObject() # devrait s'appeler getObjects...
except KeyError:
continue
images += object_to_images(root_objects)
return images
if __name__ == "__main__":
fichier = sys.argv[1]
images = pdf_to_images(fichier)
#print("n".join(map(str, images)))
for image in images:
image.show()
pass

它适用于几乎所有的PDF文件，但其中一个文件的行为很奇怪。这是一个PDF样本，可以在这里找到。当我把这个PDF文件传递给上面的代码时，结果是倾斜的，黑色和白色是颠倒的。

关于这个PDF中的图像对象，我唯一注意到的是它有一个DecodeParms条目，其值为{'/Predictor': 15, '/Columns': 2550, '/Colors': 3}。我不知道它是否相关，但这是我测试过的唯一一个有这些功能的PDF。

提前感谢

Answer 1

是的，这是相关的。数据使用FlateCode进行编码，为预测器。因此，如果您需要Image对象的原始像素，则需要在使用zlib.decompress之后对数据进行去预测。请参阅PNG规范了解如何做到这一点。

PyPDF2可能还应该为您提供一些可用于所有这些的方法。