基于改进YOLOv8的朝天椒轻量级检测方法
投稿时间:2024-05-09  修订日期:2024-05-21  点此下载全文
引用本文:
摘要点击次数: 82
全文下载次数: 0
作者单位邮编
李瑞 贵州大学机械工程学院 550025
张富贵* 贵州大学 机械工程学院 550025
吴雪梅 贵州大学 机械工程学院 
袁奎 贵州大学机械工程学院 
郑乐 贵州大学 机械工程学院 
李鑫 贵州大学 机械工程学院 
基金项目:贵州省科技创新基地建设项目《山地智能农业装备重点实验室建设》(黔合中引地[2023]010)
中文摘要:为提高朝天椒果实的检测精度,实现将模型方便快速部署到移动端,提出了一种基于改进YOLOv8的轻量化朝天椒果实检测方法。在骨干网络中使用GhostConv和全新设计的C2fGhost替换传统卷积与C2f模块,,减少了网络参数和计算量,提升了模型的检测性能;为提高模型在复杂背景下朝天椒果实的检测效果,添加全局注意力机制模块于颈部网络,提高模型的特征融合能力;使用SIoU边界损失函数替代原损失函数,提升了网络边界框回归性能和对小目标果实的检测效果。实验结果表明,对自然环境下建立的朝天椒果实数据集,改进后模型的参数量、计算量和权重大小分别为2.46×106、10.5GFLOPS和6.02MB,仅为原始网络模型的81%、37%和96%,且检测速度和平均精度分别达到了96Frame/s和88.5%。不仅提高了检测精度,还大幅减少了模型的整体体积,可为后续朝天椒果实采摘机器人移动端的部署和应用提供参考和依据。
中文关键词:图像处理  朝天椒  YOLOv8  轻量化网络  小目标  自然环境  注意力机制  损失函数
 
Lightweight Detection of Natural Environment Clustered Pepper Based on Improved YOLOv8n
Abstract:In order to improve the detection accuracy of clustered pepper fruits and realize the convenient and rapid deployment of the model to mobile terminal, a lightweight clustered pepper fruits detection method based on improved YOLOv8 is proposed. In the backbone network, GhostConv and the newly designed C2fGhost are used to replace the traditional convolution and C2f modules, which reduces the network parameters and computation volume, and improves the detection performance of the model; in order to improve the model's effect of detecting artichoke fruits in complex backgrounds, the global attention mechanism module is added in the neck network to improve the feature fusion capability of the model; and the SIoU boundary loss function is used to replace the original loss function, which improves the model's feature fusion capability. function is used to replace the original loss function, which improves the performance of network bounding box regression and the detection effect of small target fruits. The experimental results show that for the clustered pepper fruits dataset established in the natural environment, the number of parameters, the computational volume and the weight size of the improved model are 2.46×106, 10.5GFLOPS and 6.02MB, respectively, which are only 81%, 37% and 96% of the original network model, and the detection speed and the average accuracy reach 96Frame/s and 88.5%, respectively, which not only improves the The detection accuracy and average accuracy not only improve the detection accuracy, but also significantly reduce the overall volume of the model, which can provide reference and basis for the subsequent deployment and application of clustered pepper fruits picking robot mobile.
keywords:image processing  clustered pepper  YOLOv8  lighteight network  small target  natural environment  attention mechanism  loss function
查看全文   查看/发表评论   下载pdf阅读器