update pdf_loader.py

2023-05-21 15:05:35 +08:00 · 2023-05-21 15:05:35 +08:00 · 3712eec6a9
parent 94b4599cda
commit 3712eec6a9
1 changed files with 2 additions and 1 deletions
--- a/loader/pdf_loader.py
+++ b/loader/pdf_loader.py
@ -29,7 +29,8 @@ class UnstructuredPaddlePDFLoader(UnstructuredFileLoader):
                    img_list = page.get_images()
                    for img in img_list:
                        pix = fitz.Pixmap(doc, img[0])
-
+                        if pix.n - pix.alpha >= 4:
+                            pix = fitz.Pixmap(fitz.csRGB, pix)
                        pix.save(img_name)

                        result = ocr.ocr(img_name)