English

中国农机化学报

中国农机化学报 ›› 2024, Vol. 45 ›› Issue (12): 245-250.DOI: 10.13733/j.jcam.issn.20955553.2024.12.036

• 农业智能化研究 • 上一篇    下一篇

基于YOLO-PS的马铃薯幼苗检测方法研究

郑红娜1, 2,周理想2,王强3   

  1. (1. 山西铁道职业技术学院,太原市,030013; 2. 太原理工大学电子信息与光学工程学院,太原市,030024; 3. 河南理工大学测绘与国土信息工程学院,河南焦作,454000)
  • 出版日期:2024-12-15 发布日期:2024-12-03
  • 基金资助:
    山西省高等学校科技创新项目(2019L0189);教育部产学合作协同育人项目(202002035019);教育部第二期供需对接就业育人项目定向人才培养培训项目(20230102038)

Research on detection method of potato seedling based on YOLO-PS

Zheng Hongna1, 2, Zhou Lixiang2, Wang Qiang3   

  1. (1. Shanxi Railway Vocational and Technical College, Taiyuan, 030013, China; 2. College of Electronic Information and Optimal Engineering, Taiyuan University of  Technology, Taiyuan, 030024, China; 3. School of Surveying, Mapping and  Land Information Engineering, Henan University of Technology, Jiaozuo, 450003, China)
  • Online:2024-12-15 Published:2024-12-03

摘要:

针对马铃薯幼苗检测中的关键挑战,提出一种基于YOLO-PS的目标检测模型。该模型在检测骨干中引入MobileNetV4-backbone以增强对不同状态幼苗的特征提取能力,并在检测头中引入DLKA注意力机制,从而增强模型对马铃薯幼苗局部特征的提取和关注。为优化边界框的精确定位,采用Focal Loss损失函数。利用Pyqt5设计马铃薯幼苗识别系统的交互界面,使其操作简便且可靠。通过试验验证,YOLO-PS模型在马铃薯幼苗检测任务中表现优异,在测试集上的精确率达到94.75%,召回率为95.58%,平均精确度均值高达96.67%。该模型在马铃薯幼苗检测中的有效性和优越性,也为类似作物的幼苗检测提供新方法。

关键词: 深度学习, 马铃薯幼苗, YOLO, 图像处理

Abstract:

Aiming at the key challenges in potato seedling detection, this paper proposes a target detection model based on YOLO-PS (You Only Look Once with Pyramid Seedling). In this model, MobileNetV4-backbone was introduced into the detection backbone to enhance the feature extraction capability of seedlings in different states. At the same time, the DLKA attention mechanism was introduced into the detection head to enhance the model's ability to extract and focus on the local features of potato seedlings in one step. In order to optimize the precise positioning of the bounding box, the Focal Loss function was used as the loss function of the model, and finally Pyqt5 was used to design a convenient and reliable interactive interface for the potato seedling identification system. YOLO-PS model was experimentally verified to exhibit excellent performance in the potato seedling detection task. On the test set, the precision  of the model reached 94.75%, the recall  was 95.58%, and the mean average precision  was as high as 96.67%. It effectively proved the effectiveness and superiority of the model in potato seedling detection. This study not only provides a reliable technical means for automated monitoring of potato seedlings, but also provides new ideas and methods for seedling detection of similar crops.

Key words: deep learning, potato seedlings, YOLO, image processing

中图分类号: