Development of the ocr part of AOI
Samo Penic
2018-11-16 02e0f7bc51acfa06e7299919b54b56a3c7eca02b
Ocr.py
@@ -1,17 +1,18 @@
from pyzbar.pyzbar import decode
from sid_process import enhanceSID
from sid_process import getSID
import cv2
import numpy as np
import math
class Paper:
    def __init__(self, filename=None):
    def __init__(self, filename=None, sid_classifier=None):
        self.filename = filename
        self.invalid = None
        self.QRData = None
        self.errors = []
        self.warnings = []
        self.sid_classifier=sid_classifier
        if filename is not None:
            self.loadImage(filename)
            self.runOcr()
@@ -213,5 +214,13 @@
            self.answerMatrix.append(oneline)
    def get_enhanced_sid(self):
        es= enhanceSID(self.img[int(0.04*self.imgHeight):int(0.08*self.imgHeight), int(0.7*self.imgWidth):int(0.99*self.imgWidth)])
        cv2.imwrite("enhancedSID.png",es)
        if self.sid_classifier is None:
            return "x"
        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,
        )
        return es