-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathRoll_no.py
49 lines (40 loc) · 1.38 KB
/
Roll_no.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
import cv2
import numpy as np
def roll_no(roll):
orginal=np.uint8(roll)
orginal=cv2.resize(orginal,(487,581))
h,w = orginal.shape
crop= orginal[165:h-15,25:w-25]
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
roll_no = []
for y in range(0, w1,np.uint(np.floor(w1/7))):
if (y +int(w1/7) > w1):
break
column = binary[0:h1, y: y +int(w1/7)]
visc=crop[0:h1, y: y +int(w1/7)]
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", row)
# cv2.waitKey(0)
_,cnts, _ = cv2.findContours(row, cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)
if len(cnts) > 0:
if cv2.contourArea(cnts[0])>80:
roll_no.append(str(count))
# cv2.imshow("Foreground", visr )
# cv2.waitKey(0)
if len(roll_no)>0:
roll_no=['0' if x == '10' else x for x in roll_no]
roll_no=''.join(roll_no)
roll_no=int(roll_no)
else:
roll_no=""
return roll_no