-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy patharea.py
46 lines (39 loc) · 1.21 KB
/
area.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
import cv2
import random
scale=0.5
circles=[]
counter=0
counter2=0
point1=[]
point2=[]
myPoints=[]
myColor=[]
def mousePoints(event,x,y,flags,params):
global counter,point1,point2,counter2,circles,myColor
if event==cv2.EVENT_LBUTTONDOWN:
if counter==0:
point1=int(x//scale),int(y//scale)
counter +=1
myColor=(random.randint(0,2)*200,random.randint(0,2)*200,random.randint(0,2)*200)
elif counter ==1:
point2=int(x//scale),int(y//scale)
type=input('Enter Type')
name=input('Enter Name')
myPoints.append([point1,point2,type,name])
counter=0
circles.append([x,y,myColor])
counter2 +=1
img=cv2.imread('main_template.jpg')
img=cv2.resize(img,(0,0),None,scale,scale)
#h,w,c =img.shape
#img=cv2.resize(img,(w//5,h//5))
while True:
#displaying points
for x,y,color in circles:
cv2.circle(img,(x,y),3,color,cv2.FILLED)
#img=cv2.resize(img,(w//5,h//5))
cv2.imshow("Image",img)
cv2.setMouseCallback("Image",mousePoints)
if cv2.waitKey(1) & 0xFF == ord('s'):
print(myPoints)
break