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

0 vote238 views1 answer
Scikit image 0.16.2
Python
0 vote254 views1 answer
Error: The file/path provided (flaskr) does not appear to exist. Please verify the path is correct. If app is not on PYTHONPATH, ensure the extension is .py
Python Flask
0 vote191 views1 answer
How to find the frequency of a wave python
Python
0 vote190 views1 answer
Label with list comprehension python
Python
0 vote282 views1 answer
Typecasting python
Python
0 vote182 views1 answer
Python remove blanks from string
Python
0 vote156 views1 answer
Pyttsx3 install
Python
0 vote207 views1 answer
How to let something see on screen with python
Python
0 vote189 views1 answer
Django less than and greater than
Python
0 vote196 views1 answer
Django edit model without loading from db
Python