ソースを参照

fix: reduce maximum image size

- Decrease the maximum width and height from 9000 to 4500 pixels
- This change aims to prevent excessive resource usage when rendering PDFs
myhloli 11 ヶ月 前
コミット
b0529b6fbd
1 ファイル変更2 行追加2 行削除
  1. 2 2
      magic_pdf/data/utils.py

+ 2 - 2
magic_pdf/data/utils.py

@@ -20,8 +20,8 @@ def fitz_doc_to_image(doc, dpi=200) -> dict:
     mat = fitz.Matrix(dpi / 72, dpi / 72)
     mat = fitz.Matrix(dpi / 72, dpi / 72)
     pm = doc.get_pixmap(matrix=mat, alpha=False)
     pm = doc.get_pixmap(matrix=mat, alpha=False)
 
 
-    # If the width or height exceeds 9000 after scaling, do not scale further.
-    if pm.width > 9000 or pm.height > 9000:
+    # If the width or height exceeds 4500 after scaling, do not scale further.
+    if pm.width > 4500 or pm.height > 4500:
         pm = doc.get_pixmap(matrix=fitz.Matrix(1, 1), alpha=False)
         pm = doc.get_pixmap(matrix=fitz.Matrix(1, 1), alpha=False)
 
 
     img = Image.frombytes('RGB', (pm.width, pm.height), pm.samples)
     img = Image.frombytes('RGB', (pm.width, pm.height), pm.samples)