Procházet zdrojové kódy

fix: wrong color space conversion

忘忧北萱草 před 1 rokem
rodič
revize
e908a3aac4
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1 1
      pdf2zh/high_level.py

+ 1 - 1
pdf2zh/high_level.py

@@ -164,7 +164,7 @@ def extract_text_to_fp(
         caching=not disable_caching,
     ), total=total_pages, position=0):
         pix = doc_en[page.pageno].get_pixmap()
-        image = np.fromstring(pix.samples, np.uint8).reshape(pix.height, pix.width, 3)
+        image = np.fromstring(pix.samples, np.uint8).reshape(pix.height, pix.width, 3)[:, :, ::-1]
         page_layout=model.predict(
             image,
             imgsz=int(pix.height/32)*32,