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

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

Research on 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 problems of low recognition accuracy and weak generalization ability for tobacco leaf curing status under the conditions of gradual color change and overlapping vein occlusion, this paper proposes TOC-YOLO, a recognition model for tobacco leaf curing status in bulk curing barns based on the improved YOLOv11n.The model enhances the extraction of tobacco leaf morphological features by introducing the VanillaNet module; embeds the SENetV2 module and the BiFormer attention mechanism to strengthen the perception of color features and capture of spatial features of tobacco leaves, improving the model’s ability to understand complex spatial structures and occlusion relationships; adopts the HS-FPN module to achieve multi-scale feature fusion, enhancing adaptability to the dynamic morphological changes of tobacco leaves during curing; and integrates the ASFF module to reduce the misjudgment rate of adjacent curing statuses.The results show 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 are 96.9%, 96.1%, 96.5% and 98.2% respectively, and its comprehensive performance is superior to comparative models such as YOLOv11n. 2) Under the same experimental conditions, each improvement measure promotes the performance improvement of the TOC-YOLO model, achieving favorable experimental results.3) In the recognition of various curing statuses, the average F1-score of the TOC-YOLO model is 96.5%, and the average mAP0.50~0.95 is 98.2%, indicating strong adaptability to different curing statuses.4) The TOC-YOLO model exhibits 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 are 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.

     

/

返回文章
返回