From 762a5e258a90387922d6c6eb3ecc9a7ca7c96144 Mon Sep 17 00:00:00 2001
From: Samo Penic <samo.penic@gmail.com>
Date: Fri, 16 Nov 2018 18:53:49 +0000
Subject: [PATCH] refactored and debugged.

---
 Ocr.py |    8 ++++++--
 1 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/Ocr.py b/Ocr.py
index 35c6729..970fb70 100644
--- a/Ocr.py
+++ b/Ocr.py
@@ -6,13 +6,14 @@
 
 
 class Paper:
-    def __init__(self, filename=None, sid_classifier=None):
+    def __init__(self, filename=None, sid_classifier=None, settings=None):
         self.filename = filename
         self.invalid = None
         self.QRData = None
+        self.settings = settings
         self.errors = []
         self.warnings = []
-        self.sid_classifier=sid_classifier
+        self.sid_classifier = sid_classifier
         if filename is not None:
             self.loadImage(filename)
             self.runOcr()
@@ -216,11 +217,14 @@
     def get_enhanced_sid(self):
         if self.sid_classifier is None:
             return "x"
+        if self.settings is not None:
+            sid_mask=self.settings.get("sid_mask", None)
         es = getSID(
             self.img[
                 int(0.045 * self.imgHeight) : int(0.085 * self.imgHeight),
                 int(0.7 * self.imgWidth) : int(0.99 * self.imgWidth),
             ],
             self.sid_classifier,
+            sid_mask
         )
         return es

--
Gitblit v1.9.3