forked from docs/doc-exports
Sheng, Lichang
86871abfe4
Reviewed-by: Pruthi, Vineet <vineet.pruthi@t-systems.com> Co-authored-by: Sheng, Lichang <lichangsheng1@noreply.gitea.eco.tsi-dev.otc-service.com> Co-committed-by: Sheng, Lichang <lichangsheng1@noreply.gitea.eco.tsi-dev.otc-service.com>
64 lines
2.4 KiB
HTML
64 lines
2.4 KiB
HTML
<a name="ocr_01_0125"></a><a name="ocr_01_0125"></a>
|
|
|
|
<h1 class="topictitle1">How Do I Convert a PDF to an Image?</h1>
|
|
<div id="body0000001705142482"><pre class="screen" id="ocr_01_0125__screen189571911163910"># -*- coding: utf-8 -*-
|
|
import os
|
|
import base64
|
|
import fitz
|
|
import io
|
|
from PIL import Image
|
|
from glob import glob
|
|
|
|
class CovertPdfToJpg:
|
|
def __init__(self, file_path, save_root):
|
|
self.file_path = file_path
|
|
self.save_root = save_root
|
|
|
|
@staticmethod
|
|
def open_pdf(file):
|
|
return fitz.open(file)
|
|
|
|
@staticmethod
|
|
def get_trans(doc, page, min_side=0, max_side=0, rotate=0.0):
|
|
""" Create a scale object. """
|
|
region = doc[page].rect
|
|
scale = 1
|
|
if max_side > min_side > 0:
|
|
scale = min_side / min(region.width, region.height)
|
|
if max(region.width, region.height) * scale > max_side:
|
|
scale = max_side / max(region.width, region.height)
|
|
trans = fitz.Matrix(scale, scale).preRotate(rotate)
|
|
return trans
|
|
|
|
def page2pix(self, doc, page, trans):
|
|
""" Parse the current page as image data based on given parameters."""
|
|
# Obtain the PDF format of a specified page. Note that page parameters need to be pre-parsed to avoid any issues.
|
|
return doc[page].getPixmap(matrix=trans, alpha=False)
|
|
|
|
def pdf_to_jpg(self, width=1024, height=1400):
|
|
doc = self.open_pdf(self.file_path)
|
|
save_dir = os.path.join(self.save_root)
|
|
if not os.path.exists(save_dir):
|
|
os.makedirs(save_dir)
|
|
print("document", len(doc), doc.pageCount)
|
|
for i in range(len(doc)):
|
|
trans = self.get_trans(doc, i, width, height, rotate=0)
|
|
try:
|
|
pdf = self.page2pix(doc, i, trans)
|
|
except:
|
|
continue
|
|
image = pdf.getPNGData()
|
|
image = Image.open(io.BytesIO(image))
|
|
print(os.path.join(
|
|
save_dir, os.path.basename(self.file_path).replace('.pdf', '') + '_' + str(i + 1) + '.jpg'))
|
|
image.save(
|
|
os.path.join(save_dir, os.path.basename(self.file_path).replace('.pdf', '') + '_' + str(i + 1) + '.jpg'))
|
|
return</pre>
|
|
</div>
|
|
<div>
|
|
<div class="familylinks">
|
|
<div class="parentlink"><strong>Parent topic:</strong> <a href="ocr_01_0046.html">APIs</a></div>
|
|
</div>
|
|
</div>
|
|
|