
改进的 YOLOv5(加入 SE 注意力模块)在大型飞机图像目标检测中的应用数据集
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本研究基于YOLOv5框架,引入SE注意力机制,显著提升了对大型飞机图像中特定目标的识别精度与效率,适用于复杂背景下的精准检测任务。
项目包含基于YOLOV5的大型飞机图像目标检测数据集实战项目,包括代码、数据集及训练好的权重参数。经测试,该项目可以直接使用,并对原始YOLOV5代码进行了改进,如增加了注意力机制SE模块等优化措施。
数据集中仅有一类目标:airplane。整个项目的大小为135MB,在经过了100个epoch的迭代后,保存于runs目录下的训练结果中最佳精度map0.5达到了0.95,而map0.5:0.95则为0.71。在训练过程中生成了验证集的混淆矩阵、PR曲线和F1曲线等。
在runs/detect目录下,项目还保存了网络推理训练集的所有结果,并且具有良好的推理效果。此外,该项目还包括检测类别的txt字典文件以供参考使用。
为了便于查看数据,提供了一个可视化py文件,用户可以随机传入一张图片来绘制边界框并将其保存在当前目录中。该脚本无需进行任何更改即可直接运行。
全部评论 (0)
还没有任何评论哟~


