이재빈

전처리 모듈 추가

1 +import cv2
2 +import numpy as np
3 +import matplotlib as plt
4 +import time
5 +fname = "./croppedimg/"
6 +index = 0
7 +prevTime = 0
8 +def filter_img(img):
9 + #이미지의 RGB값을 분석하여 찾는 실내 Tag가 맞는지 판별
10 + img = cv2.resize(img, (10,10))
11 + first = [0,0,0]
12 + for x_loc in range(0, 10):
13 + for y_loc in range(0, 10):
14 + bgr_value = img[x_loc,y_loc]
15 + first=first+bgr_value
16 + first[0] = first[0]/100
17 + first[1] = first[1]/100
18 + first[2] = first[2]/100
19 + blue = first[0]<200 and first[0]>120
20 + green = first[1]>120 and first[1]<210
21 + red = first[2]>130 and first[2]<230
22 +
23 + if(blue and green and red):
24 + return True
25 + else:
26 + return False
27 +def bboxes(inp,prevTime):
28 + #Frame을 인자로 전달받음
29 + img = inp
30 + start = time.time()
31 + curTime = time.time()
32 + # img2gray = cv2.imread(fname,0)
33 + # img = cv2.namedWindow(img,cv2.WINDOW_NORMAL)
34 + # img = cv2.resizeWindow(img,600,600)
35 + img_final = inp
36 + # img_final = cv2.namedWindow(fname,cv2.WINDOW_NORMAL)
37 + # img_final = cv2.resizeWindow(fname,600,600)
38 + img2gray = cv2.cvtColor(inp, cv2.COLOR_BGR2GRAY) #GRAY Image 8bit per pixel
39 + ret, mask = cv2.threshold(img2gray, 180, 255, cv2.THRESH_BINARY) #threshold : distinguish background, object
40 + image_final = cv2.bitwise_and(img2gray, img2gray, mask=mask) #bitwise
41 + ret, new_img = cv2.threshold(img_final, 180, 255, cv2.THRESH_BINARY) # Nfor black text , cv.THRESH_BINARY_IV
42 + newimg = cv2.cvtColor(new_img, cv2.COLOR_BGR2GRAY) #Gray Image converting
43 + #newimg = cv2.GaussianBlur(newimg, (3,3),0)
44 +
45 + # remove noise from image
46 + #kernel = cv2.getStructuringElement(cv2.MORPH_CROSS, (5,1))
47 + # to manipulate the orientation of dilution , large x means horizonatally dilating more, large y means vertically dilating more
48 + #dilated = cv2.dilate(newimg, kernel, iterations=1) # dilate
49 + # erode = cv2.erode(newimg, kernel)
50 + _,contours, _ = cv2.findContours(newimg, cv2.RETR_CCOMP, cv2.CHAIN_APPROX_NONE) # get contours
51 + #cv2.CHAIN_APPROX_NONE: 모든 컨투어 포인트를 반환
52 + for contour in contours:
53 + # get rectangle bounding contour
54 + [x, y, w, h] = cv2.boundingRect(contour)
55 + # remove small false positives that aren't textq
56 + # text인식하기. width, height
57 + #if w > 50 or h > 35 or w<25:
58 + #continue
59 + if h / w > 1.0 or w / h > 2.0:
60 + continue
61 + #if h>40 or w>70:
62 + #continue
63 + if y>150:
64 + continue
65 + cropped = img_final[y :y + h , x : x + w]
66 + # draw rectangle around contour on original image
67 + if(filter_img(cropped)):
68 + cv2.rectangle(img, (x, y), (x + w, y + h), (0, 0, 255), 3)
69 + cv2.putText(img,"cropped", (x-50,y-10), cv2.FONT_HERSHEY_COMPLEX_SMALL, 1, (0,0,255), 1)
70 + #cropped = img_final[y :y + h , x : x + w]
71 + else:
72 + continue
73 + img = cv2.resize(img, (720, 380))
74 + sec = curTime - prevTime
75 + prevTime = curTime
76 + fps = 1/(sec)
77 + #print ("Time {0} ".format(sec))
78 + #print ("Estimated fps {0} ".format(fps))
79 + str1 = ("FPS : {0}".format(int(fps)))
80 + cv2.putText(img, str1, (0, 40), cv2.FONT_HERSHEY_COMPLEX_SMALL, 0.8, (0, 255, 0),1)
81 + cv2.imshow('captcha_result', img)
82 + return prevTime
83 +
84 +cap = cv2.VideoCapture(0) #동영상 파일 읽어옴
85 +while (cap.isOpened()):
86 + ret, inp = cap.read() #프레임을 읽어옴, 읽어온 프레임을 인자로 bboxes 전달
87 + if(ret): #success boolean
88 + prevTime = bboxes(inp, prevTime)
89 + if cv2.waitKey(1) & 0xFF == ord('q'):
90 + print("Terminate Process..")
91 + break
92 +cap.release() #파일 닫아줌
...\ No newline at end of file ...\ No newline at end of file