Remove alpha channel from LA images

Calamari-OCR · Feb 25, 2022 · 29903c3 · 29903c3
1 parent dfe453c
commit 29903c3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/calamari_ocr/utils/image.py b/calamari_ocr/utils/image.py
@@ -40,6 +40,8 @@ def load_image(self, image_path: str) -> np.ndarray:
 
         if img_channels == self.params.channels:
             pass  # good
+        elif img_channels == 2 and self.params.channels == 1:
+            img = img[:, :, 0]
         elif img_channels == 3 and self.params.channels == 1:
             if self.params.to_gray_method == "avg":
                 img = np.mean(img.astype("float32"), axis=-1).astype(dtype=img.dtype)