计算机视觉无处不在-从面部识别，制造，农业到自动驾驶汽车。今天，我们将通过动手实践进入现代计算机视觉世界，学习如何使用YOLO算法检测车牌。

来自Pexels的mali maeder的照片应用于定制的YOLO车牌检测模型
传统计算机视觉方法使用vision方法进行检测。但由于阈值和轮廓检测的局限性，其算法在部分图像上有效，但无法推广。通过本次学习，我们将拥有可以在任何天气情况下用于检测车牌的强大模型。

数据收集与准备

我们有一个可靠的数据库，其中包含数百张汽车图像，但是在网上共享它是不道德的。因此，小伙伴必须自己收集汽车图像。我们将使用如下照片进行演示和验证：

我们还应该收集各种光照条件下的车牌图像，并从不同角度拍摄图像。在完成数据收集之后，我们将使用一个名为LabelIMG的免费工具来完成这项工作。
cd labelImg-master
brew install qt
brew install libxml2
make qt5py3
python labelImg.py
从终端执行这些操作将打开此窗口：

图2-启动LabelIMG
接下来，单击左侧菜单上的“打开目录”图标。找到存储汽车图像的文件夹。它会自动打开第一个图像：

图像3 -用LabelIMG打开图像（作者提供的图像）
左侧面板中的标签会显示YOLO。接下来，按键盘上的W键以打开RectBox 工具。在车牌周围绘制一个矩形，输入标签，然后单击“确定”：

图4 -在板周围绘制矩形
按CTRL + S将板块坐标保存到文本文件。单个文件应如下所示：

图像5- LabelIMG生成的文本文件
该LabelIMG软件将保存的矩形框坐标文件对应于保存的每个图像。还将所有类的列表保存到名为classes.txt的文件中。我们打开它，其中可能列出了许多我们不感兴趣的内容。可以删除license-plate以外的所有内容。

图6-类列表
仍然存在一个问题。我们只有一个类（license-plate），但是坐标文件中的类索引为15，我们只需查看图片5并自行验证即可。

我们将使用Python加载每个坐标文件，以通过将类索引更改为1来解决此问题。这是代码段：
import glob

Contains all .txt files except our listof classes

txt_files = [file for file inglob.glob(‘images/*.txt’) if file != ‘images/classes.txt’]

Read every .txt file and store it’scontent into variable curr

for file in txt_files:
with open(file, ‘r’) as f:
curr = f.read()

Replace class index 15 with 1 and storeit in a variable new

new = curr.replace('15 ', '1 ')

Once again open every .txt file and makethe replacement

for file in txt_files:
with open(file, ‘w’) as f:
f.write(new)
到目前为止，我们已经完成了数据集收集和准备。接下来，我们需要压缩文件并进行模型训练。

模型训练

我们已经有几百个带有标签的汽车图像。足够我们训练一个好的YOLO模型，接下来就是我们要做的。我们将在带有GPU后端的Google Colab上训练模型。我们的案例中，在Colab中训练模型大约需要2个小时，但是时间会有所变化，具体取决于GPU和数据集的大小。

我们将通过接下来的七个步骤来训练模型。

第1步-配置GPU环境
在新的笔记本中，转到运行时-更改运行时类型，然后在下拉列表中选择GPU：

图7-切换到GPU运行时
第2步-挂载Google云端硬盘
在Google云端硬盘中，创建一个备份文件夹。我们将其命名为yolo-license-plates。这就是存储模型权重和配置的地方。在第一个单元格中，执行以下代码来安装Google云端硬盘：
from google.colab import drive
drive.mount(’/content/gdrive’)
!ln -s /content/gdrive/My\ Drive/ /mydrive
步骤3 —下载并配置Darknet
Darknet是一个开源神经网络框架，具有YOLO对象检测系统。我们可以通过执行以下代码行来下载它：
！git clonehttps://github.com/AlexeyAB/darknet
接下来，在darknet/Makefile中我们必须配置一些内容。在以下行上更改值：
第1行-从GPU=0到GPU=1
第2行-从CUDNN=0到CUDNN=1
第4行-从OPENCV=0到OPENCV=1
并保存文件。这些更改使我们可以在训练时使用GPU。我们现在可以编译Darknet
%cd darknet
!make
这需要等待几分钟，我们在编译完成后继续进行下一步。

步骤4 —配置设置文件
要知道如何设置YOLO配置文件，我们需要知道有多少个类。我们只有一个— license-plate，但这可能会根据我们正在处理的问题的类型而改变。

接下来，我们需要计算批次数和过滤器数。以下是计算公式：
批次=类数* 2000
过滤器=（类别数+ 5）* 3
在我们的例子中，值分别为2000和18。为了准确起见，请复制YOLO配置文件：
！cp cfg / yolov3.cfg cfg / yolov3-train.cfg
并在cfg/yolov3-train.cfg中进行以下更改：
第3行-从batch=1到batch=64
第4行-从subdivisions=1到subdivisions=16
第20行-从max_batches=500200到max_batches=2000
603、689和776行-从filters=255到filters=18
610、696和783行-从classes=80到classes=1
保存文件。接下来，我们必须创建两个文件-data/obj.names和data/obj.data。这些文件包含有关类名和备份文件夹的信息：
!echo -e ‘license-plate’ >data/obj.names
!echo -e ‘classes = 1\ntrain =data/train.txt\nvalid = data/test.txt\nnames = data/obj.names\nbackup =/mydrive/yolo-license-plates’ > data/obj.data
到现在，我们已经完成了配置，现在让我们上传并准备图像。

步骤5-上传和解压缩图像
我们只需将zip文件拖放到“文件”侧边栏菜单中即可。完成后应如下所示：

图8-zip文件上传后的Colab文件菜单
下一步是为图像创建一个文件夹并将其解压缩：
！mkdir数据/对象
！unzip …/images.zip -d data / obj
现在data/obj文件夹应包含图像及其各自的文本文件。

步骤6-训练准备
接下来，我们要做的就是创建一个data/train.txt文件。它将包含所有训练图像的路径：
import glob
images_list = glob.glob(‘data/obj/*.jpg’)
with open(‘data/train.txt’, ‘w’) as f:
f.write(’\n’.join(images_list))
最后，我们必须下载预训练的Darknet卷积网络：
！wgethttps://pjreddie.com/media/files/darknet53.conv.74
下载将花费几秒钟，但是一旦完成，我们便可以进入到模型训练阶段。

步骤7 —模型训练
现在，开始训练过程可以归结为一行shell代码：！./ darknet检测器火车数据/obj.datacfg / yolov3-train.cfg darknet53.conv.74 -dont_show

图9-Colab中的YOLO模型训练
现在，我们需要等待。训练过程可能需要几个小时，具体取决于图像的数量。权重每10到15分钟自动保存到您的Google云端硬盘备份文件夹中。在下一部分中，我们将创建一个脚本，用于在图像上查找和标记车牌。

模型评估

模型训练完成后，我们应该在Google云端硬盘备份文件夹中拥有三个文件：
权重文件— yolov3_training_final.weights
配置文件- yolov3_testing.cfg
类文件— classes.txt
将它们下载到计算机上的单个文件夹中，然后打开JupyterLab，我们可以从LicensePlateDetector下面的代码段中复制：
import cv2
import numpy as np
import matplotlib.pyplot as plt
class LicensePlateDetector:
def init(self, pth_weights: str, pth_cfg: str, pth_classes: str):
self.net = cv2.dnn.readNet(pth_weights, pth_cfg)
self.classes = []
with open(pth_classes, ‘r’) as f:
self.classes = f.read().splitlines()
self.font = cv2.FONT_HERSHEY_PLAIN
self.color = (255, 0, 0)
self.coordinates = None
self.img = None
self.fig_image = None
self.roi_image = None

def detect(self, img_path: str):
orig = cv2.imread(img_path)
self.img = orig
img = orig.copy()
height, width, _ = img.shape
blob = cv2.dnn.blobFromImage(img, 1 / 255, (416, 416), (0, 0, 0),swapRB=True, crop=False)
self.net.setInput(blob)
output_layer_names = self.net.getUnconnectedOutLayersNames()
layer_outputs = self.net.forward(output_layer_names)
boxes = []
confidences = []
class_ids = []

   for output in layer_outputs:for detection in output:scores = detection[5:]class_id = np.argmax(scores)confidence = scores[class_id]if confidence > 0.2:center_x = int(detection[0]* width)center_y = int(detection[1]* height)w = int(detection[2] *width)h = int(detection[3] *height)x = int(center_x - w / 2)y = int(center_y - h / 2)boxes.append([x, y, w, h])confidences.append((float(confidence)))class_ids.append(class_id)indexes = cv2.dnn.NMSBoxes(boxes, confidences, 0.2, 0.4)if len(indexes) > 0:for i in indexes.flatten():x, y, w, h = boxes[i]label = str(self.classes[class_ids[i]])confidence =str(round(confidences[i],2))cv2.rectangle(img, (x,y), (x +w, y + h), self.color, 15)cv2.putText(img, label + ' ' +confidence, (x, y + 20), self.font, 3, (255, 255, 255), 3)self.fig_image = imgself.coordinates = (x, y, w, h)return

def crop_plate(self):
x, y, w, h = self.coordinates
roi = self.img[y:y + h, x:x + w]
self.roi_image = roi
return
此类有两种方法：
detect(img_path)–用于从输入图像中检测车牌并在其周围绘制一个矩形。crop_plate()–用于从图像中裁剪检测到的车牌。如果您想应用一些OCR来提取文本，则此方法可用。
lpd = LicensePlateDetector(
pth_weights=‘yolov3_training_final.weights’,
pth_cfg=‘yolov3_testing.cfg’,
pth_classes=‘classes.txt’
)

Detect license plate

lpd.detect(‘001.jpg’)

Plot original image with rectangle aroundthe plate

plt.figure(figsize=(24, 24))
plt.imshow(cv2.cvtColor(lpd.fig_image, cv2.COLOR_BGR2RGB))
plt.savefig(‘detected.jpg’)
plt.show()

Crop plate and show cropped plate

lpd.crop_plate()
plt.figure(figsize=(10, 4))
plt.imshow(cv2.cvtColor(lpd.roi_image,cv2.COLOR_BGR2RGB))
上面的代码片段构成了LicensePlateDetector该类的一个实例，检测到车牌，并将其裁剪。这是可视化输出：

图10- YOLO模型和LicensePlateDetector类的评估
我们可以在过去几个小时（或几天）内完成的所有工作。YOLO模型可以完美运行，并且可以用于任何使用案例。

结论

这是一篇很长的分享。祝贺小伙伴一次坐下来就可以完成实践。我们也是花了几天的时间才能了解YOLO的工作原理以及如何制作自定义对象检测器。使用相同的方法，小伙伴们可以构建任何类型的对象检测器。例如，我们重复使用具有不同文本标签的相同图像来检测汽车颜色和汽车品牌。期待小伙伴们的大显身手。

YOLOV3实现车牌检测相关推荐

基于yolov3的目标检测与LPRnet字符识别的车牌识别（CCPD2020新能源车牌数据集）
文章目录前言一.程序思路二.使用步骤 1.配置环境 2.文件结构 3.准备数据集 4.训练有问题欢迎指正前言项目放这:车牌识别基于python和pytorch平台,使用CCPD2020新 ...
车牌识别，车辆检测，车牌检测和识别，与车相关的点点滴滴
前言: 最近接到了几个单子,都是和车辆有关的,其中有车辆检测,还有车牌检测和识别等,与车相关的,这里就是总结下自己所了解到的相关技术,同时备份到自己的网盘将代码. 车辆检测车辆检测,车辆检测是一个老 ...
移动端开源车牌检测识别项目
点击上方"码农的后花园",选择"星标" 公众号精选文章,第一时间送达今天给大家推荐一个由智云视图车牌识别的开源项目HyperLPR,HyperLPR是一个基 ...
使用深度学习进行自动车牌检测和识别
作者 | 小白来源 | 小白学视觉介绍在现代世界的不同方面,信息技术的大规模集成导致了将车辆视为信息系统中的概念资源.由于没有任何数据,自主信息系统就没有任何意义,因此需要在现实和信息系统之间改 ...
【深度学习】使用深度学习进行自动车牌检测和识别
介绍在现代世界的不同方面,信息技术的大规模集成导致了将车辆视为信息系统中的概念资源.由于没有任何数据,自主信息系统就没有任何意义,因此需要在现实和信息系统之间改革车辆信息.这可以通过人工代理或特殊智 ...
【camera】基于深度学习的车牌检测与识别系统实现（课程设计）
基于深度学习的车牌检测与识别系统实现(课程设计) 代码+数据集下载地址:下载地址用python3+opencv3做的中国车牌识别,包括算法和客户端界面,只有2个文件,surface.py是界面代码, ...
智慧工地-基于深度学习yolov3的安全帽检测以及安全帽检测数据集（1）
基于深度学习yolov3的安全帽检测以及安全帽检测数据集数据集和代码下载地址:下载地址数据可视化: 根目录下运行命令: show_yolo_anno.py (注意脚本内相关参数配置 ) 模型训练 ...
从YOLOv1到YOLOv3，目标检测的进化之路
本文来自 CSDN 网站,作者 EasonApp. 作者专栏: http://dwz.cn/7ZGrif YOLOv1 这是继 RCNN,fast-RCNN 和 faster-RCNN之后,Ross ...
车牌检测识别--Towards End-to-End Car License Plates Detection and Recognition with Deep Neural Networks
Towards End-to-End Car License Plates Detection and Recognition with Deep Neural Networks https://ar ...

YOLOV3实现车牌检测