-
Notifications
You must be signed in to change notification settings - Fork 0
/
RedMarlboro_RECOGNITION.py
50 lines (39 loc) · 1.36 KB
/
RedMarlboro_RECOGNITION.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
50
import cv2
import numpy as np
MATCH_RATE=27
detector=cv2.xfeatures2d.SIFT_create()
ZERO=0
flannParam=dict(algorithm=ZERO,tree=5)
flann=cv2.FlannBasedMatcher(flannParam,{})
trainImg=cv2.imread("marlboro.png",0)
trainKP,trainDesc=detector.detectAndCompute(trainImg,None)
cam=cv2.VideoCapture(0)
while True:
ret, QueryImgBGR=cam.read()
QueryImg=cv2.cvtColor(QueryImgBGR,cv2.COLOR_BGR2GRAY)
queryKP,queryDesc=detector.detectAndCompute(QueryImg,None)
matches=flann.knnMatch(queryDesc,trainDesc,k=2)
MatchPoint=[]
for m,n in matches:
if(m.distance<0.75*n.distance):
MatchPoint.append(m)
if(len(MatchPoint)>MATCH_RATE):
tp=[]
qp=[]
for m in MatchPoint:
tp.append(trainKP[m.trainIdx].pt)
qp.append(queryKP[m.queryIdx].pt)
tp,qp=np.float32((tp,qp))
H,status=cv2.findHomography(tp,qp,cv2.RANSAC,3.0)
h,w=trainImg.shape
trainBorder=np.float32([[[0,0],[0,h-1],[w-1,h-1],[w-1,0]]])
queryBorder=cv2.perspectiveTransform(trainBorder,H)
cv2.polylines(QueryImgBGR,[np.int32(queryBorder)],True,(0,255,0),5)
#Here you can specify what you want to heppen when the Marlboro is placed
print ("success")
break
cv2.imshow('LocateIMG',QueryImgBGR)
if cv2.waitKey(10)==ord('q'):
break
cam.release()
cv2.destroyAllWindows()