本资源提供基于YOLOv8的全面车辆及车牌检测解决方案,包括详尽的源代码、部署指南、预训练模型及性能评估图表。
基于YOLOv8的车辆+车牌检测系统源码(部署教程、训练好的模型及各项评估指标曲线).zip
平均准确率:0.96
类别:car、plate
【资源介绍】
1、ultralytics-main
该文件夹包含YOLOv8源代码,涵盖分类、目标检测、姿态估计和图像分割四个部分。我们使用的是detect部分,即目标检测相关代码。
2、搭建环境
在Windows系统、macOS系统及Linux系统中均可运行。
安装Anaconda 和 PyCharm 后,在 Anaconda 中新建一个名为 YOLOv8 的虚拟空间:
```
conda create -n YOLOv8 python==3.8
```
创建完成后,激活该环境:
```
source activate YOLOv8 # Linux或macOS系统使用此命令
activate YOLOv8 # Windows系统使用此命令
```
接着,在虚拟环境中安装requirements.txt文件中的所有依赖包。
若希望加速安装过程,请考虑更换到清华源等国内镜像。
3、训练模型
进入ultralytics-main/ultralytics/yolov8/detect目录,其中datasets为准备好的数据集。data目录下的bicycle.yaml是本人用于自行车检测的数据配置文件,您需要根据具体需求修改或创建新的yaml文件。
train.py中的第238行需改为`data = cfg.data or path/to/your/data.yaml`(例如:./data/carplate.yaml)
在同文件的第237行指定预训练模型。如果有可用显卡,可以在第239行设置设备号,如四张显卡配置为:`args = dict(model=model, data=data, device=0,1,2,3)`
完成上述步骤后运行train.py开始训练。
训练结束后,在runsdetect文件夹下会生成包含模型和评估指标的train*文件夹。
4、推理测试
在predict.py中,第87行需将`model = cfg.model or yolov8n.pt`中的yol8vn.pt替换为刚才训练完成后的模型路径(位于ultralytics-main/ultralytics/yolov8/detect/runsdetect目录下)。
待测图片或视频放置于assets文件夹中,运行predict.py进行测试。结果将保存在runsdetect/train文件夹内。