label_parking_lot.py

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()