|
基于改进YOLOv8的虫草小目标检测算法
|
Abstract:
针对青藏高原特殊环境下虫草检测面临的复杂高原背景、目标遮挡频繁、虫草形态细长且易与自然背景混淆等挑战,本文基于YOLOv8模型提出改进方法。首先,在可变卷积(Deformable Convolution)的基础上设计双层可变卷积(Double-layer Deformable Convolution),建立双层动态卷积调整机制,利用特征偏移量自适应调整卷积核的大小和形状,提高特征饱和度,缓解下采样带来的信息失衡。其次,针对虫草因遮挡导致的漏检问题,融合空间增强注意力机制(SEAM, Spatially Enhanced Attention Module),通过深度可分离卷积和残差模块增强未遮挡部分的语义特征,优化空间通道中的权重信息,有效提升模型对遮挡环境下的信息提取和检测能力。最后,引入新的检测头FASFF-head,以自适应学习多尺度特征图的空间权重,进行空间特征融合,确保多尺度特征的协调性,且在原有检测层之上添加小目标专用检测层,使得在高密草丛环境下,虫草的检测精度得到显著提升。以上实验表明,改进模型在自建藏区虫草数据集上的mAP@0.5和mAP@0.5:0.95对比YOLOv8模型分别提升4.2%和2.9%;在Flavia Dataset公开数据集上的实验结果可以发现,YOLOv8-DSEAM 除了参数量略高于YOLOv10n,mAP@0.5比YOLOv10n提高了1.3%,mAP@0.5:0.95比YOLOv10n提高了0.8%,充分地展现了改进后的模型在高密草丛场景下的检测力和泛化力。
To address the challenges of caterpillar fungus detection in the complex plateau environment of the Qinghai-Tibet Plateau, including intricate high-altitude backgrounds, frequent target occlusion, and the elongated morphology of cordyceps that easily blends with natural surroundings, this paper proposes improvements based on the YOLOv8 model. First, we design a Double-layer Deformable Convolution building upon Deformable Convolution, establishing a dual-layer dynamic convolution adjustment mechanism. This utilizes feature offsets to adaptively adjust convolution kernel size and shape, enhancing feature saturation and alleviating information imbalance caused by downsampling. Second, to tackle missed detection due to occlusion, we integrate the Spatially Enhanced Attention Module (SEAM). Through depthwise separable convolution and residual modules, this enhances semantic features of unoccluded regions and optimizes weight information in spatial channels, effectively improving information extraction and detection capabilities in occluded environments. Finally, we introduce a novel FASFF-head detection head to adaptively learn spatial weights of multi-scale feature maps for spatial feature fusion, ensuring multi-scale feature coordination. Additionally, a dedicated small-target detection layer is added above the original detection layers, significantly improving detection accuracy in dense grassland environments. Experimental results demonstrate that the improved model achieves 4.2% and 2.9% increases in mAP@0.5 and mAP@0.5:0.95 respectively compared to YOLOv8 on our self-built Tibetan Cordyceps dataset. On the public Flavia Dataset, YOLOv8-DSEAM shows superior performance: while slightly higher in parameters than YOLOv10n, it improves
[1] | 辛梦茹, 谢地, 毛维东, 等. 冬虫夏草繁育品保健功能的多组学研究[J]. 中国医药工业杂志, 2025, 56(2): 205-216. |
[2] | 谭志. 基于深度学习的目标检测与识别技术[M]. 北京: 化学工业出版社, 2021. |
[3] | 米增, 连哲. 面向通用目标检测的YOLO方法研究综述[J]. 计算机工程与应用, 2024, 60(21): 38-54. |
[4] | 刘勇, 唐霞. 计算机视觉技术在智慧农业中的应用[J]. 中国农业资源与区划, 2024, 45(12): 84. |
[5] | 于庆旭, 欧阳安, 陈永生, 等. 西北道地中药材种植及生产机械化概况[J]. 中国农机化学报, 2025, 46(1): 290-298. |
[6] | 尹芹, 方晖, 王金东, 等. 增强小目标检测性能的通道自注意力机制算法研究[J]. 南京邮电大学学报(自然科学版), 2022, 42(4): 69-74. |
[7] | 彭林聪, 王克瑞, 周浩, 等. 基于改进SSD的目标检测算法[J]. 激光杂志, 2024, 45(11): 71-76. |
[8] | 夏子林, 张新洲, 王文波, 等. 基于机器视觉的蚕豆荚高精度检测方法研究[J]. 中国农机化学报, 2025, 46(1): 157-163. |
[9] | 吴金山, 黄子建, 孙嘉华, 等. 基于改进EfficientDet算法的珍稀植物识别[J]. 林业科技通讯, 2024(12): 30-35. |
[10] | 林晨, 钟雅露, 孔彦琪, 等. 基于改进RT-DETR的草莓叶枯病实时检测[J]. 信息技术与信息化, 2025(1): 79-82. |
[11] | 谢家兴, 廖飞, 王卫星, 等. 基于改进Faster R-CNN的荔枝病虫害检测[J]. 华中农业大学学报, 2025, 44(1): 62-73. |
[12] | 付晓鸽, 李涵, 左治江, 等. 基于Mask R-CNN的复杂环境下辣椒识别方法研究[J]. 中国农机化学报, 2024, 45(9): 215-219. |
[13] | 王元昊, 娄欢欢, 罗红品, 等. 基于改进YOLOv8算法对被遮挡柑橘的识别与定位优化[J]. 西南大学学报(自然科学版), 2025, 47(2): 171-183. |
[14] | 陈江川, 曹芷怡, 余赠宇, 等. 基于YOLOv5的农田病虫害检测识别方法[J]. 南方农机, 2025, 56(4): 112-117. |
[15] | 承达瑜, 赵伟, 何伟德, 等. 基于改进YOLOv5n模型的农作物病虫害识别方法[J]. 江苏农业学报, 2024, 40(11): 2021-2031. |
[16] | 许韬. 无人机图像识别算法在林业领域检测的应用[J]. 农业技术与装备, 2024(11): 3-5. |
[17] | 陈金荣, 许燕, 周建平, 等. 基于YOLO-SSAR的自然环境下红花检测算法[J]. 农业工程学报, 2025, 41(2): 215-223. |
[18] | 李涛, 买买提明∙艾尼, 古丽巴哈尔∙托乎提, 等. 基于改进YOLOv8的小棚架下无核白葡萄果梗识别[J]. 中国农机化学报, 2025, 46(2): 259-263, 278. |
[19] | 毕泽洋, 杨立伟, 吕树盛, 等. 基于EDH-YOLO的轻量型温室番茄检测方法[J]. 农业机械学报, 2024, 55(z2): 246-254. |
[20] | 李兆东, 章艳芳, 汪蕴红, 等. 基于YC-YOLO v7模型的油菜幼苗株数识别方法[J]. 农业机械学报, 2024, 55(12): 322-332. |
[21] | 黄友锐, 王小桥, 韩涛, 等. 基于改进YOLO v8n的甜菜杂草检测算法研究[J]. 江苏农业科学, 2024, 52(24): 196-204. |
[22] | 刘梦姝, 张春琪, 晁金阳, 等. 基于YOLO v8n改进的小麦病害检测系统[J]. 农业机械学报, 2024, 55(z1): 280-287, 355. |
[23] | Wu, S.G., Bao, F.S., Xu, E.Y., Wang, Y., Chang, Y.-F. and Xiang, Q.-L. (2007) A Leaf Recognition Algorithm for Plant Classification Using Probabilistic Neural Network. 2007 IEEE International Symposium on Signal Processing and Information Technology, Giza, 15-18 December 2007, 11-16. https://doi.org/10.1109/ISSPIT.2007.4458016 |