
改进的 YOLOv5(加入 SE 注意力模块)在大型行人图像目标检测中的应用数据集
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
本研究通过引入SE注意力机制增强YOLOv5模型,在大型行人图像数据集中显著提升了目标检测性能,尤其增强了对复杂背景中行人的识别精度。
项目包含基于YOLOV5的大型人图像目标检测数据集的目标检测实战项目,包括代码、数据集以及训练好的权重参数。经过测试,代码可以直接使用。
对YOLOV5进行了改进,加入了注意力机制SE模块,并进行了一系列其他优化。
数据集介绍:这是一个单类别的行人检测数据集。
类别分为1种:person
【yolov5】项目总大小为135MB
在训练过程中迭代了100个epoch,在runs目录下保存了训练结果,最佳精度map0.5为0.86,map0.5:0.95为0.54。此外,还生成了验证集的混淆矩阵、PR曲线和F1曲线等数据。
在runsdetect目录中保存了网络推理训练集的所有结果,并且推理效果良好。
项目还包括检测类别的txt字典文件以及用于可视化的py脚本,可以直接运行并绘制边界框。
全部评论 (0)
还没有任何评论哟~


