Extract image from pdf python

import PyPDF2

from PIL import Image

if __name__ == '__main__':
    input1 = PyPDF2.PdfFileReader(open("input.pdf", "rb"))
    page0 = input1.getPage(0)
    xObject = page0['/Resources']['/XObject'].getObject()

    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"

            if xObject[obj]['/Filter'] == '/FlateDecode':
                img = Image.frombytes(mode, size, data)
                img.save(obj[1:] + ".png")
            elif xObject[obj]['/Filter'] == '/DCTDecode':
                img = open(obj[1:] + ".jpg", "wb")
                img.write(data)
                img.close()
            elif xObject[obj]['/Filter'] == '/JPXDecode':
                img = open(obj[1:] + ".jp2", "wb")
                img.write(data)
                img.close()

Source: stackoverflow.com

Similar codes

3 votes345 views3 answers
String list to list
Python
0 vote342 views1 answer
Convert 2.9k to 2900 in python
Python
1 vote435 views1 answer
Image delete in django from the folder
Python Django
11 votes332 views6 answers
Linear regression python code
Python
0 vote352 views1 answer
Use python3.7 as default
Python
47 votes380 views5 answers
Sort the dictionary in python
Python
0 vote358 views2 answers
How to input a string character into a numpy zeros imatrix n python
Python
4 votes360 views4 answers
Interface graphique python
Python
45 votes370 views6 answers
Python webbrowser
Python
23 votes348 views4 answers
Python delete white spaces
Python