mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-01-18 22:44:02 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

work on digits_video.py

Browse Source
This commit is contained in:
Alexander Mordvintsev 2012-06-27 09:46:04 +00:00
parent abde1377f3
commit c71c82a8f1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
samples/python2/digits_video.py
View File

@ -31,14 +31,14 @@ def main():
boxes = [] boxes = []
for cnt in contours: for cnt in contours:
x, y, w, h = cv2.boundingRect(cnt) x, y, w, h = cv2.boundingRect(cnt)
if h < 20 or h > 60 or 1.2*h < w: if h < 16 or h > 60 or 1.2*h < w:
continue continue
cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 255, 0)) cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 255, 0))
sub = bin[y:,x:][:h,:w] sub = bin[y:,x:][:h,:w]
#sub = ~cv2.equalizeHist(sub) #sub = ~cv2.equalizeHist(sub)
#_, sub_bin = cv2.threshold(sub, 0, 255, cv2.THRESH_BINARY_INV | cv2.THRESH_OTSU) #_, sub_bin = cv2.threshold(sub, 0, 255, cv2.THRESH_BINARY_INV | cv2.THRESH_OTSU)
s = 1.1*h/SZ s = float(h)/SZ
m = cv2.moments(sub) m = cv2.moments(sub)
m00 = m['m00'] m00 = m['m00']
if m00/255 < 0.1*w*h or m00/255 > 0.9*w*h: if m00/255 < 0.1*w*h or m00/255 > 0.9*w*h:
@ -49,7 +49,7 @@ def main():
c0 = np.float32([SZ/2, SZ/2]) c0 = np.float32([SZ/2, SZ/2])
t = c1 - s*c0 t = c1 - s*c0
A = np.zeros((2, 3), np.float32) A = np.zeros((2, 3), np.float32)
A[:,:2] = np.eye(2)*2 A[:,:2] = np.eye(2)*s
A[:,2] = t A[:,2] = t
sub1 = cv2.warpAffine(sub, A, (SZ, SZ), flags=cv2.WARP_INVERSE_MAP | cv2.INTER_LINEAR) sub1 = cv2.warpAffine(sub, A, (SZ, SZ), flags=cv2.WARP_INVERSE_MAP | cv2.INTER_LINEAR)
sub1 = digits.deskew(sub1) sub1 = digits.deskew(sub1)