计算机视觉结合深度学习项目-智能停车场空车位实时识别_大厦智能停车场空余车位算法

    if make_copy:
        new_image = np.copy(image)
    #Step 1: 过滤部分直线
    cleaned = []
    for line in lines:
        for x1,y1,x2,y2 in line:
            if abs(y2-y1) <=1 and abs(x2-x1) >=25 and abs(x2-x1) <= 55:
                cleaned.append((x1,y1,x2,y2))

首先我们还是过滤掉一些直线。

import operator
list1 = sorted(cleaned, key=operator.itemgetter(0, 1))

对于这十二列，每一列的左上角的坐标点我们是可以得到x1-x12的我们要对这些列进行一次排序操作。让计算机识别出哪一列是第一列，哪一列是第十二列。

    clusters = {}
    dIndex = 0
    clus_dist = 10

    for i in range(len(list1) - 1):
        distance = abs(list1[i+1][0] - list1[i][0])
        if distance <= clus_dist:
            if not dIndex in clusters.keys(): clusters[dIndex] = []
            clusters[dIndex].append(list1[i])
            clusters[dIndex].append(list1[i + 1]) 

        else:
            dIndex += 1

这里就是做了一下对于所有排序好的直线进行了一个归类操作，把哪些直线归为一列。并且进行添加。直到把每一列都进行分出来。

    rects = {}
    i = 0
    for key in clusters:
        all_list = clusters[key]
        cleaned = list(set(all_list))#一列中的所有直线的坐标信息
        if len(cleaned) > 5:
            cleaned = sorted(cleaned, key=lambda tup: tup[1])#对直线进行排序
            avg_y1 = cleaned[0][1]#这个对于一列来说是固定的
            avg_y2 = cleaned[-1][1]#这个对于一列来说是固定的
            avg_x1 = 0
            avg_x2 = 0
            for tup in cleaned:
                avg_x1 += tup[0]
                avg_x2 += tup[2]
            avg_x1 = avg_x1/len(cleaned)
            avg_x2 = avg_x2/len(cleaned)
            rects[i] = (avg_x1, avg_y1, avg_x2, avg_y2)
            i += 1
    
    print("Num Parking Lanes: ", len(rects))

然后我们对每一列进行操作，把每一列的每一个车位的所有坐标信息提取出来。然后再通过得到的坐标及进行画出来这个矩形。

    buff = 7#微调数值
    for key in rects:
        tup_topLeft = (int(rects[key][0] - buff), int(rects[key][1]))
        tup_botRight = (int(rects[key][2] + buff), int(rects[key][3]))
        cv2.rectangle(new_image, tup_topLeft,tup_botRight,(0,255,0),3)
    return new_image, rects

![在这里插入图片描述](https://img-blog.csdnimg.cn/8e8