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

AI科技大本营
2022年1月20日05时

作者 | 努比

来源 | 小白学视觉

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

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

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

1. 选择停车位的坐标并将其保存到文件中。

2. 从文件中获取坐标，并确定该点是否可用。

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

为了使这一过程尽可能简单，从现在开始，我们将这两个脚本称为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

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

1. 打开image变量中的视频流；suc确定流是否成功打开。

2. 将第一帧写入frame0.jpg。

3. 流被释放，所有窗口都关闭。

4. 新保存的图片将以img变量形式读取。

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函数的参数如下：

1. “选择器”是允许我们选择投资回报率的窗口的名称。

2. img是包含我们要选择图像的变量。

3. showCrosshair = Flase删除选区内部的中心线。可以将其设置为True，因为对结果没有影响。

4. fromCenter = False是一个非常重要的参数，因为如果将其设置为True，则正确的选择会困难得多。

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

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

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

detector探测器

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

1. 从.csv文件获取坐标。

2. 从中构建新图像。

3. 应用OpenCV中可用的Canny函数。

4. 计算新图像内的白色像素。

5. 建立一个点内的像素范围将被占用。

6. 在实时供稿上绘制一个红色或绿色矩形。

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

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