extract image from pdf python

1import PyPDF2
2
3from PIL import Image
4
5if __name__ == '__main__':
6    input1 = PyPDF2.PdfFileReader(open("input.pdf", "rb"))
7    page0 = input1.getPage(0)
8    xObject = page0['/Resources']['/XObject'].getObject()
9
10    for obj in xObject:
11        if xObject[obj]['/Subtype'] == '/Image':
12            size = (xObject[obj]['/Width'], xObject[obj]['/Height'])
13            data = xObject[obj].getData()
14            if xObject[obj]['/ColorSpace'] == '/DeviceRGB':
15                mode = "RGB"
16            else:
17                mode = "P"
18
19            if xObject[obj]['/Filter'] == '/FlateDecode':
20                img = Image.frombytes(mode, size, data)
21                img.save(obj[1:] + ".png")
22            elif xObject[obj]['/Filter'] == '/DCTDecode':
23                img = open(obj[1:] + ".jpg", "wb")
24                img.write(data)
25                img.close()
26            elif xObject[obj]['/Filter'] == '/JPXDecode':
27                img = open(obj[1:] + ".jp2", "wb")
28                img.write(data)
29                img.close()
30