Merge branch 'master' of ssh://git.iqser.com:2222/rr/image-prediction

image_prediction/image_extractor/extractors/parsable.py

@@ -13,8 +13,11 @@ from image_prediction.image_extractor.extractor import ImageExtractor, ImageMeta
 from image_prediction.info import Info
 from image_prediction.stitching.stitching import stitch_pairs
 from image_prediction.stitching.utils import validate_box_coords, validate_box_size
+from image_prediction.utils import get_logger
 from image_prediction.utils.generic import lift
 
+logger = get_logger()
+
 
 class ParsablePDFImageExtractor(ImageExtractor):
     def __init__(self, verbose=False, tolerance=0):
@@ -162,7 +165,11 @@ def has_alpha_channel(doc, xref):
     if maybe_smask:
         return any([doc.extract_image(maybe_smask) is not None, bool(fitz.Pixmap(doc, maybe_smask).alpha)])
     else:
-        return bool(fitz.Pixmap(doc, xref).alpha)
+        try:
+            return bool(fitz.Pixmap(doc, xref).alpha)
+        except ValueError:
+            logger.debug(f"Encountered invalid xref `{xref}` in {doc.metadata.get('title', '<no title>')}.")
+            return False
 
 
 def tiny(metadata):

src/serve.py

@@ -10,7 +10,6 @@ from image_prediction.utils.banner import show_banner
 
 
 def main():
-    logger = get_logger()
 
     def predict(pdf):
         # Keras service_estimator.predict stalls when service_estimator was loaded in different process;
@@ -26,7 +25,9 @@ def main():
 
 
 if __name__ == "__main__":
-    logging.basicConfig(level=CONFIG.service.logging_level)
+    logging.basicConfig()
+    logger = get_logger()
+    logger = logger.setLevel(CONFIG.service.logging_level)
 
     show_banner()