栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

训练好的YoloV5模型进行目标检测

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

训练好的YoloV5模型进行目标检测

接上一篇博客,本篇记录第一次训练的检测效果
训练得到的最优模型在...runstrainexp6weightsbest.pt位置

激活yolov5py38环境,进入yolov5-master文件夹

输入以下命令,使用视频来检查模型训练的效果

python detect.py --source D:yoloimageprocessmovie2021_1229_132828_422.MP4 --weights runstrainexp6weightsbest.pt --conf-thres 0.25

--source 后输入检测的源文件路径,--weights后输入训练好的模型权重文件
--conf-thres 是置信度阈值,低于该值的检测结果将被忽略。

源文件可以是多种格式,支持图像和视频,包括:'bmp', 'jpg', 'jpeg', 'png', 'tif', 'tiff', 'dng', 'webp', 'mpo';'mov', 'avi', 'mp4', 'mpg', 'mpeg', 'm4v', 'wmv', 'mkv'。

训练完成后结果如下,平均每帧检测时间1.3ms

输出的视频存储在runsdetectexp中

效果如下


第二次训练调整了batch_size=32,再次进行训练,受限于GPU显存不足,完成训练后无法进行验证,单独运行val.py对第二次训练的模型进行验证,设置数据为mydataset.yaml,权重为刚刚训练好的best.pt

python val.py --data D:yoloyolov5-masterdatamydataset.yaml --weights D:yoloyolov5-masterrunstrainexp8weightsbest.pt

训练好的验证结果存储在runs/val/exp中

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/755012.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号