
关于改进版YOLO-v5s在植物识别中的应用研究——以轻量化模型为例- 马宏兴
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本文由马宏兴撰写,探讨了改进版YOLO-v5s算法在植物识别领域的应用,并通过实例分析展示了轻量级模型的有效性。
为了便于调查宁夏全区荒漠草原植物种类及其分布情况,有必要对植物识别方法进行深入研究。鉴于YOLO v5s模型参数量较大,并且在复杂背景下难以准确识别植物的问题,我们提出了一种改进的轻量化模型——YOLO v5s-CBD。
该模型通过引入带有Transformer模块的主干网络BoTNet(瓶颈变换器网络),使卷积操作与自注意力机制相结合,从而增强了感受野。此外,在特征提取过程中融入了坐标注意力(Coordinate attention, CA),这有助于有效捕捉通道和位置之间的关系,并进一步提升模型的特征提取能力。同时,引入SIoU函数用于计算回归损失,解决了预测框与真实框不匹配的问题;并且使用深度可分离卷积(Depthwise separable convolution, DSC)来减小模型体积。
实验结果显示,在一块Nvidia GTX A5000 GPU上,改进后的YOLO v5s-CBD模型的帧率达到140帧/秒,模型大小为8.9MB。其精确率P达到95.1%,召回率R为92.9%,综合评价指标F1值为94.0%;平均精度均值mAP则达到了95.7%。此外,在VOC数据集上该模型的平均精度均值可达80.09%。
全部评论 (0)
还没有任何评论哟~


