基于python和OpenCV构建智能停车系统

点击上方“小白学视觉”，选择加"星标"或“置顶”

重磅干货，第一时间送达

当今时代最令人头疼的事情就是找不到停车位，尤其是找20分钟还没有找到停车位。

根据复杂性和效率的不同，任何问题都具有一个或多个解决方案。目前智能停车系统的解决方案，主要包括基于深度学习实现，以及基于重量传感器、光传感器实现等。

本期我们将一起通过使用摄像头和少量代码来实现最简单的智能停车系统。该解决方案所使用的概念非常简单。它由具有以下两个脚本组成：

将该解决方案分成两个脚本的原因是，避免在每次确定是否有可用停车位的时候，就进行停车位的选择。

为了使这一过程尽可能简单，从现在开始，我们将这两个脚本称为selector和detector。

相关依赖

在本文中，我们使用python 3.7.6，但其他版本（例如3.6或3.8）当然也可以使用。首先我们要检查python的版本，我们通过在控制台中编写python –version，即可返回已安装的python版本。

C:\Users\Razvan>python --versionPython 3.7.6

在开始构建该系统依赖项之前，我们可以设置一个虚拟环境。通过以下链接我们可以了解更多有关虚拟环境的信息https://docs.python.org/3.7/tutorial/venv.html。

也可以使用conda创建和管理环境。有关更多信息见https://docs.anaconda.com/anaconda/。

在python中设置完所有内容后， 最重要的依赖关系将是OpenCV库。通过pip将其添加到虚拟环境中，可以运行pip install opencv-python。

要检查所有设置是否正确，我们可以使用以下cv2.__version__命令打印环境中可用的当前OpenCV版本。

(OpenCV) C:\Users\Razvan>pythonPython 3.7.6 (default, Jan  8 2020, 20:23:39) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32Type "help", "copyright", "credits" or "license" for more information.>>> import cv2>>> print(cv2.__version__)4.2.0>>>

在第一行中，我们可以看到在该项目中使用了名为OpenCV的虚拟环境。

步骤

首先，我们需要安装一个停车场摄像头。由于我们并没有一个窗户可以看到的任何停车场，因此我们选择使用旧汽车玩具和印刷纸。另外，我在停车场上方设置了一个网络摄像头，以获取良好的图像，因此我们正在处理的图像如下所示：

selector选择器

接下来，我们来介绍编码部分。首先，我们需要构建选择器。我们从导入所需模块开始

import cv2import csv

之后，我们开始获取图像，在该图像上选择停车位。为此，我们可以选择摄网络摄像头提供的第一帧，保存并使用该图像选择停车位。下面的代码是这样的：

VIDEO_SOURCE = 1
cap = cv2.VideoCapture(VIDEO_SOURCE)suc, image = cap.read()cv2.imwrite("frame0.jpg", image)cap.release()cv2.destroyAllWindows()img = cv2.imread("frame0.jpg")

现在，我们已经保存了第一帧并在img变量中将其打开，可以使用selectROIs函数标记停车位。ROI被定义为感兴趣的区域，代表图像的一部分，我们将在其上应用不同的函数以及滤波器来获取结果。

返回到selectROIs函数，这将返回一个列表（类型：numpy.ndarray），其中包含我们组装图像所需的数字及其边界ROI。

r = cv2.selectROIs('Selector', img, showCrosshair = False, fromCenter = False)

我们的列表将保存在变量r中。赋予cv2.selectROIs函数的参数如下：

选择所有停车位之后，是时候将它们写入.csv文件了。为此，我们需要将r变量转换为python列表，可以使用rlist = r.tolist()命令实现。

拥有适当的数据后，我们将其保存到.csv文件中，以备将来使用。

with open('data/rois.csv', 'w', newline='') as outf:  csvw = csv.writer(outf)  csvw.writerows(rlist)

detector探测器

现在我们已经选择了停车位，是时候进行一些图像处理了。解决这个问题的方法如下：

对于所有这些操作，我们需要定义一个要应用于每个位置的函数。该函数如下所示：

def drawRectangle(img, a, b, c, d):    sub_img = img[b:b + d, a:a + c]    edges = cv2.Canny(sub_img, lowThreshold, highThreshold)    pix = cv2.countNonZero(edges)
    if pix in range(min, max):        cv2.rectangle(img, (a, b), (a + c, b + d), (0, 255, 0), 3)        spots.loc += 1    else:        cv2.rectangle(img, (a, b), (a + c, b + d), (0, 0, 255), 3)

现在我们已经实现了所需的功能，如果我们直接将其应用于.csv文件中的每组坐标效果可能并不好。因此我们做如下处理

首先，我们的有一些参数如果可以在脚本运行时（也可以在通过GUI）实时调整它们，那就更好了。为此，我们需要构建一些轨迹栏。OpenCV为我们提供这项功能。

我们需要一个回调函数，该函数不执行任何操作，但作为使用OpenCV创建轨迹栏的参数是必需的。实际上，回调参数具有明确定义的用途，但我们在此不使用它。要了解有关此内容的更多信息，查阅OpenCV文档。

def callback(foo):    pass

现在我们需要创建轨迹栏。我们将使用cv2.namedWindow和cv2.createTrackbar功能。

cv2.namedWindow('parameters')cv2.createTrackbar('Threshold1', 'parameters', 186, 700, callback)cv2.createTrackbar('Threshold2', 'parameters', 122, 700, callback)cv2.createTrackbar('Min pixels', 'parameters', 100, 1500, callback)cv2.createTrackbar('Max pixels', 'parameters', 323, 1500, callback)

现在，我们已经创建了用于操作参数的GUI，只剩下一件事了。这就是图像中可用斑点的数量。在drawRectangle中定义为spot.loc。这是一个静态变量，必须在程序开始时进行定义。该变量为静态变量的原因是，我们希望调用的每个drawRectangle函数都将其写入相同的全局变量，而不是每个函数都使用一个单独的变量。这样可以防止返回的可用空间数量大于实际的可用空间数量。

为了实现这一点，我们只需要使用它的loc静态变量创建spots类。

class spots:    loc = 0

现在我们已经准备就绪，只需要从.csv文件中获取数据，将其所有数据转换为整数，然后在无限循环中应用构建的函数即可。

with open('data/rois.csv', 'r', newline='') as inf:    csvr = csv.reader(inf)    rois = list(csvr)
rois = [[int(float(j)) for j in i] for i in rois]VIDEO_SOURCE = 1cap = cv2.VideoCapture(VIDEO_SOURCE)
while True:    spots.loc = 0
    ret, frame = cap.read()    ret2, frame2 = cap.read()    min = cv2.getTrackbarPos('Min pixels', 'parameters')    max = cv2.getTrackbarPos('Max pixels', 'parameters')    lowThreshold = cv2.getTrackbarPos('Threshold1', 'parameters')    highThreshold = cv2.getTrackbarPos('Threshold2', 'parameters')
    for i in range(len(rois)):        drawRectangle(frame, rois[i][0], rois[i][1], rois[i][2], rois[i][3])
    font = cv2.FONT_HERSHEY_SIMPLEX    cv2.putText(frame, 'Available spots: ' + str(spots.loc), (10, 30), font, 1, (0, 255, 0), 3)    cv2.imshow('Detector', frame)
    canny = cv2.Canny(frame2, lowThreshold, highThreshold)    cv2.imshow('canny', canny)
    if cv2.waitKey(1) & 0xFF == ord('q'):        break
cap.release()cv2.destroyAllWindows()

拓展

在我们的循环中实际上只是调用的构建函数要复杂一点。

首先，我们将空间的数量初始化为0，以防止每帧添加数字。

其次，我们进入两个处理流以显示真实图像和已处理的图像。这有助于更好地了解此脚本的工作方式以及图像的处理方式。

然后，我们需要在每次迭代中获取我们创建的参数 GUI中的参数值。这是通过cv2.getTrackbarPos功能完成的。

接下来最重要的部分，将drawRectangle函数应用到Selector脚本获取的所有坐标上。

最后，在结果图像上写下可用斑点的数量，显示Canny函数的结果，显然，这是一种众所周知的停止循环的方法。

我们现在便完成了一个智能停车项目！

总结

如今，智能停车已成为热门话题之一，并且有许多实现方式可以导致良好的功能系统。我们这处理方法并不是完美的，有许多方法可以更好地优化结果，并且可以在更多情况下使用。但是，即使这不能解决停车场危机，也可能是导致危机 的主要原因。

交流群

欢迎加入公众号读者群一起和同行交流，目前有SLAM、三维视觉、传感器、自动驾驶、计算摄影、检测、分割、识别、医学影像、GAN、算法竞赛等微信群（以后会逐渐细分），请扫描下面微信号加群，备注：”昵称+学校/公司+研究方向“，例如：”张三 + 上海交大 + 视觉SLAM“。请按照格式备注，否则不予通过。添加成功后会根据研究方向邀请进入相关微信群。请勿在群内发送广告，否则会请出群，谢谢理解~