Skip to content

Instantly share code, notes, and snippets.

@catwhocode

catwhocode/extract-image.py

Created October 31, 2024 03:27
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save catwhocode/d7f7016000b371551fd9d1879d2691c0 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save catwhocode/d7f7016000b371551fd9d1879d2691c0 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. catwhocode renamed this gist Oct 31, 2024. 1 changed file with 0 additions and 0 deletions.
    0 extract.py → extract-image.py
    View file
    File renamed without changes.
  2. catwhocode created this gist Oct 31, 2024.
    33 changes: 33 additions & 0 deletions extract.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,33 @@
    # Install package first:
    # pip install PyMuPDF Pillow

    import fitz
    import io
    from PIL import Image

    file = "source.pdf"

    pdf_file = fitz.open(file)

    for page_index in range(len(pdf_file)):
    page = pdf_file.load_page(page_index) # load the page
    image_list = page.get_images(full=True) # get images on the page

    if image_list:
    print(f"[+] {len(image_list)} images on page {page_index}")
    else:
    print("[!] No image on page", page_index)

    for image_index, img in enumerate(image_list, start=1):
    xref = img[0]

    base_image = pdf_file.extract_image(xref)
    image_bytes = base_image["image"]

    image_ext = base_image["ext"]

    image_name = f"image{page_index+1}_{image_index}.{image_ext}"

    with open(image_name, "wb") as image_file:
    image_file.write(image_bytes)
    print(f"[+] saved as {image_name}")