Update pdf_to_img.py

周伟奇
Showing 1 changed file with 6 additions and 1 deletions
pdf_to_img.py
--- a/pdf_to_img.py
View file @7ef3754
+++ b/pdf_to_img.py
View file @7ef3754
@@ -81,7 +81,12 @@ 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
+            # return doc.extractImage(xref)
    @staticmethod
    def get_img_data(pix):