-
Notifications
You must be signed in to change notification settings - Fork 5
/
label_parking_lot.py
81 lines (47 loc) · 1.76 KB
/
label_parking_lot.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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
import cv2
print("[INFO] Loading parking lots ...")
img = cv2.imread("parking_lot.png")
print("[INFO] Parking lots loaded successfully!")
print("[NOTE] Please enter the parking lot lable as shown on the screen (yellow area)")
file = open("parking_labels.txt","r+")
file.truncate(0)
file.close()
file = open("parking_area_coordinates.txt")
lines = file.readlines()
lines = [line.strip() for line in lines]
total_parking_lots = len(lines)
parking_lot_coords = list()
for i in range(len(lines)):
coords = lines[i].split()
left = int(coords[0])
top = int(coords[1])
right = int(coords[2])
bottom = int(coords[3])
coords = [left, top, right, bottom]
parking_lot_coords.append(coords)
for i in range(len(parking_lot_coords)):
if i > 0:
cv2.rectangle(img=img,
pt1=(parking_lot_coords[i][0], parking_lot_coords[i][1]),
pt2=(parking_lot_coords[i][2], parking_lot_coords[i][-1]),
color=(0,255,255),
thickness=2)
cv2.rectangle(img=img,
pt1=(parking_lot_coords[i-1][0], parking_lot_coords[i-1][1]),
pt2=(parking_lot_coords[i-1][2], parking_lot_coords[i-1][-1]),
color=(0,255,0),
thickness=2)
if i == 0:
cv2.rectangle(img=img,
pt1=(parking_lot_coords[i][0], parking_lot_coords[i][1]),
pt2=(parking_lot_coords[i][2], parking_lot_coords[i][-1]),
color=(0,255,255),
thickness=2)
cv2.imshow("Parking lot", img)
cv2.waitKey(10)
parking_label = input("Enter this parking label: ")
with open('parking_labels.txt', 'a') as file:
file.write("{}\n".format(parking_label))
print("[INFO] Parking {} is saved!".format(parking_label))
continue
cv2.destroyAllWindows()