OMR Reader using Python

Author: koshygeoji

Category.py

@@ -0,0 +1,37 @@
+import cv2
+import numpy as np
+
+def category(cat):
+    cate=""
+    orginal=np.uint8(cat)
+    orginal=cv2.resize(orginal,(202,626))
+    c = ["GEN","OBC1","OBC2","SC","ST","PH"]
+
+    h,w = orginal.shape
+    crop= orginal[140:h-5,140:w-5]
+#    cv2.imshow('l',crop)
+
+    h1,w1 = crop.shape
+    th, im_th = cv2.threshold(crop,127,255,0)
+
+    im_th=~im_th
+    kernel = np.ones((5,5), np.uint8) 
+    binary = cv2.erode(im_th, kernel, iterations=2) 
+    
+    count = 0
+    for x in range(0, h1,np.uint(np.floor(h1/6))):
+           
+        if (x+int(h1/6) > h1):
+               break
+        row = binary[x:x+int(h1/6),:] 
+        visr=crop[x:x+int(h1/6),:] 
+        count+=1
+    #    cv2.imshow("Foreground", visr)
+    #    cv2.waitKey(0)
+        _,cnts, _ = cv2.findContours(row, cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)  
+        
+        if len(cnts) == 1: 
+            cate=c[count-1]               
+    
+
+    return cate

Center_code.py

@@ -0,0 +1,48 @@
+import cv2
+import numpy as np
+
+def code(center):
+    
+    orginal=np.uint8(center)
+    orginal=cv2.resize(orginal,(267,583))
+
+    h,w = orginal.shape
+    crop= orginal[158:h-5,18:w-18]
+    
+    
+    th, im_th = cv2.threshold(crop,127,255,0)
+    im_th=~im_th
+    kernel = np.ones((5,5), np.uint8) 
+    binary = cv2.erode(im_th, kernel, iterations=2) 
+    
+    h1,w1 = crop.shape
+    code = []
+    for y in range(0, w1,np.uint(np.floor(w1/4))): 
+        if (y +int(w1/4) > w1):
+            break 
+        column = binary[0:h1, y: y +int(w1/4)]
+        visc=crop[0:h1, y: y +int(w1/4)]
+        count=0          
+        for x in range(0, h1,np.uint(np.floor(h1/10))):
+               
+            if (x+int(h1/10) > h1):
+                   break
+            row = column[x:x+int(h1/10),:] 
+            visr=visc[x:x+int(h1/10),:] 
+            count+=1
+            
+    #        cv2.imshow("Foreground", visr)
+    #        cv2.waitKey(0)
+            
+            _,cnts, _ = cv2.findContours(row, cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)       
+            if len(cnts) == 1: 
+                code.append(str(count))
+    if len(code)>0:
+        code=['0' if x == '10' else x for x in code]
+        code=''.join(code)
+        code=int(code)
+    else:
+        code=""
+    return code
+
+