fix pdf_to_img

周伟奇
Showing 1 changed file with 5 additions and 1 deletions
src/common/tools/pdf_to_img.py
--- a/src/common/tools/pdf_to_img.py
View file @07c844a
+++ b/src/common/tools/pdf_to_img.py
View file @07c844a
@@ -205,7 +205,11 @@ class PDFHandler:
            tpix = fitz.Pixmap(fitz.csRGB, pix)
            return tpix
        else:
-            return doc.extractImage(xref)
+            res = doc.extractImage(xref)
+            if res is None:
+                return fitz.Pixmap(doc, xref)
+            else:
+                return res
    @staticmethod
    def get_img_data(pix):