本平台为互联网非涉密平台,严禁处理、传输国家秘密或工作秘密

基于改进YOLOv11n的密集烤房烟叶烘烤状态识别模型

A recognition model for tobacco leaf curing status in bulk curing barns based on improved YOLOv11n

  • 摘要: 针对烟叶颜色渐变、叶脉堆叠遮挡情况下,对烘烤状态识别精度低和泛化能力弱的问题,提出了一种基于改进YOLOv11n的密集烤房烟叶烘烤状态识别模型TOC-YOLO。该模型通过引入VanillaNet模块增强烟叶形态特征提取;嵌入SENetV2模块与BiFormer注意力机制,强化烟叶的颜色特征感知与空间特征捕捉,提升模型对复杂空间结构与遮挡关系的理解能力;采用HS-FPN模块实现多尺度特征融合,提升对烘烤过程中烟叶动态形态变化的适应能力;通过集成ASFF模块,降低相邻烘烤状态的误判率。结果表明:①测试集中,TOC-YOLO模型的边框精度(Bp)、召回率(R)、F1分数及mAP0.50~0.95分别为96.9%、96.1%、96.5%和98.2%,综合性能优于YOLOv11n等对比模型。②在相同试验条件下,每个改进措施均促进了TOC-YOLO模型的性能提升,取得了较好的试验结果。③在各烘烤状态识别中,TOC-YOLO模型的F1分数平均值为96.5%,mAP0.50~0.95平均值为98.2%,对不同烘烤状态具有较强的适配性。④TOC-YOLO模型在变黄后期、定色后期和干筋后期等关键烘烤状态均表现出优异的特征感知能力。⑤在泛化测试集上,相较于YOLOv11n基准模型,TOC-YOLO模型的BPRF1分数和mAP0.50~0.95分别提升了5.3、2.5、4.0和0.8百分点,表明该模型具有较强的跨场景泛化使用能力。该研究为密集烤房烟叶烘烤状态识别提供了技术支持。

     

    Abstract: To address the issues of low recognition accuracy and weak generalization ability for tobacco leaf curing status due to gradual color change and vein overlapping, a recognition model TOC-YOLO for tobacco leaf curing status in bulk curing barns based on the improved YOLOv11n was proposed. The model enhanced the extraction of tobacco leaf morphological features by introducing the VanillaNet module. In addition, by embedding the SENetV2 module and the BiFormer attention mechanism to strengthen the visual perception of color features and capture of spatial features of tobacco leaves, the model's ability to understand complex spatial structures and occlusion relationships was improved. The model adopted the HS-FPN module to achieve multi-scale feature fusion, enhancing its adaptability to the dynamic morphological changes of tobacco leaves during curing, and integrated the ASFF module to reduce the misjudgment rate of adjacent curing status. The results showed that: 1) In the test set, the bounding box precision (Bp), recall (R), F1 score and mAP0.50-0.95 of the TOC-YOLO model were 96.9%, 96.1%, 96.5% and 98.2% respectively, and its comprehensive performance was superior to comparative models such as YOLOv11n. 2) Under the same experimental conditions, each of the improvement measure promoted the performance of the TOC-YOLO model, achieving favorable experimental results. 3) In the recognition of various curing status, the average F1 score of the TOC-YOLO model was 96.5%, and the average mAP0.50-0.95 was 98.2%, indicating strong adaptability to different curing status. 4) The TOC-YOLO model exhibited excellent feature perception ability in key curing stages such as the late yellowing stage, late color-fixing stage and late stem-drying stage. 5) On the generalization test set, compared with the baseline YOLOv11n model, the Bp, R, F1 score and mAP0.50-0.95 of the TOC-YOLO model were increased by 5.3, 2.5, 4.0 and 0.8 percentage points respectively, demonstrating strong cross-scenario generalization capability. This study provides technical support for the recognition of tobacco leaf curing status in bulk curing barns.

     

/

返回文章
返回