外贸推广信邮件,seo技术教程网,内网网站建设所需硬件设备,wordpress集成dz项目介绍
上一篇文章介绍了基于卷积神经网络的交通标志分类识别Python交通标志识别基于卷积神经网络的保姆级教程#xff08;Tensorflow#xff09;#xff0c;并且最后实现了一个pyqt5的GUI界面#xff0c;并且还制作了一个简单的Falsk前端网页实现了前后端的一个简单交互…项目介绍
上一篇文章介绍了基于卷积神经网络的交通标志分类识别Python交通标志识别基于卷积神经网络的保姆级教程Tensorflow并且最后实现了一个pyqt5的GUI界面并且还制作了一个简单的Falsk前端网页实现了前后端的一个简单交互只能实现单张交通标志图像的分类没有位置检测功能并且不支持视频的实时检测识别总体上来讲较为简单。本文介绍一个交通标志识别的进阶项目–基于Yolov5的交通标志检测识别它不仅能实现图片的多目标检测识别还可以实现视频的实时检测识别。大家可以看一下视频展示效果如下链接
视频演示交通标志检测视频演示
获取代码 创建虚拟环境 conda create -n yolov5 python3.8.5安装pytorch如果不会弄GPU的直接安装CPU版本
安装CPU版本torch
conda install pytorch1.8.0 torchvision0.9.0 torchaudio0.8.0 cpuonly安装GPU版本torch以我个人为例我的显卡是3060TiCUDA版本是11.7
pip install torch1.11.0cu113 torchvision0.12.0cu113 torchaudio0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113安装其他依赖库
pip install -r requirements.txt
pip install pyqt55.15.6
pip install pycocotools-windows2.0.0.2测试代码是否能跑 python detect.py --source data/images/traffic_sign.jpg --weights runs/train/exp7/weights/best.pt项目文件夹data/images/bus.jpg图片在识别前的样子 成功运行命令如上图显示信息所示识别结果存在项目文件夹runs\detect\exp7文件中
训练可忽略
接下来用pycharm打开项目然后在Terminal中操作输入以下命令
python train.py --data traffic_sign_data.yaml --cfg mask_yolov5s.yaml --weights pretrained/yolov5s.pt --epoch 100 --batch-size 4 训练比较耗时我给的项目压缩包中是已经训练好的了。所以训练的步骤可以跳过。 训练完成后模型评估结果在模型训练完成后会在runs/train目录下生成一个exp文件里面包含了训练结果以及一些评估指标。 根据输出的识别结果可以知道模型的准确率是非常的高
运行GUI界面
在经过训练之后打开windows.py这个代码直接点击运行结果如下所示
