将PDF转换为PNG

import os
from PyPDF2 import PdfFileReader
from PIL import Image

def pdf_to_png(input_pdf, output_folder):
    pdf_file = PdfFileReader(open(input_pdf, "rb"))
    num_pages = pdf_file.getNumPages()

    for i in range(num_pages):
        page = pdf_file.getPage(i)
        page_image = page.extractImage()
        xObject = page['/Resources']['/XObject']
        for obj in xObject:
            if xObject[obj]['/Subtype'] == '/Image':
                size = (xObject[obj]['/Width'], xObject[obj]['/Height'])
                data = xObject[obj].getData()
                if xObject[obj]['/ColorSpace'] == '/DeviceRGB':
                    mode = "RGB"
                else:
                    mode = "P"
                im = Image.frombytes(mode, size, data)
                im.save(os.path.join(output_folder, f"page_{i}.png"))

input_pdf = "example.pdf"
output_folder = "output"
pdf_to_png(input_pdf, output_folder)