From 9c222b2a0b151e7219e30f0145aa92872890d838 Mon Sep 17 00:00:00 2001 From: Samo Penic <samo.penic@gmail.com> Date: Wed, 21 Nov 2018 18:48:59 +0000 Subject: [PATCH] Fixes in qr code, in sid third algoritm in answer matrix finding locations... --- aoi_ocr/Ocr.py | 15 ++++++++------- 1 files changed, 8 insertions(+), 7 deletions(-) diff --git a/aoi_ocr/Ocr.py b/aoi_ocr/Ocr.py index bc7f062..e022ff1 100644 --- a/aoi_ocr/Ocr.py +++ b/aoi_ocr/Ocr.py @@ -99,7 +99,8 @@ # todo, make better tresholding def imgTreshold(self): (self.thresh, self.bwimg) = cv2.threshold( - self.img, 128, 255, cv2.THRESH_BINARY | cv2.THRESH_OTSU + self.img, 128, 255, + cv2.THRESH_BINARY | cv2.THRESH_OTSU ) def getSkewAngle(self): @@ -242,7 +243,7 @@ es, err, warn = getSID( self.img[ int(0.04 * self.imgHeight) : int(0.095 * self.imgHeight), - int(0.7 * self.imgWidth) : int(0.99 * self.imgWidth), + int(0.65 * self.imgWidth) : int(0.95 * self.imgWidth), ], self.sid_classifier, sid_mask, @@ -275,7 +276,7 @@ data = qrdata.split(",") retval = { "exam_id": int(data[1]), - "page_no": int(data[3])+1, + "page_no": int(data[3]), "paper_id": int(data[2]), "faculty_id": int(data[0]), "sid": None @@ -291,17 +292,17 @@ data["errors"] = self.errors data["warnings"] = self.warnings data["up_position"] = ( - list(self.xMarkerLocations[1] / self.imgWidth), - list(self.yMarkerLocations[1] / self.imgHeight), + list(self.xMarkerLocations[0] / self.imgWidth), + list(self.xMarkerLocations[1] / self.imgHeight), ) data["right_position"] = ( - list(self.xMarkerLocations[1] / self.imgWidth), + list(self.yMarkerLocations[0] / self.imgWidth), list(self.yMarkerLocations[1] / self.imgHeight), ) data["ans_matrix"] = ( (np.array(self.answerMatrix) > self.settings["answer_threshold"]) * 1 ).tolist() - if data["sid"] is None and data["page_no"] == 2: + if data["sid"] is None and data["page_no"] == 1: data["sid"] = self.get_enhanced_sid() output_filename=os.path.join(self.output_path, '.'.join(self.filename.split('/')[-1].split('.')[:-1])+".png") cv2.imwrite(output_filename, self.img) -- Gitblit v1.9.3