Wang Suwan, Zhao Jianqing, Cai Yucheng, Li Yan, Qi Xuerui, Qiu Xiaolei, Yao Xia, Tian Yongchao, Zhu Yan, Cao Weixing, Zhang Xiaohu
National Engineering and Technology Center for Information Agriculture, Nanjing Agricultural University, Nanjing, 210095, China.
College of Geography, Jiangsu Second Normal University, Nanjing, 211200, China.
Plant Methods. 2024 Jan 29;20(1):15. doi: 10.1186/s13007-024-01147-w.
The number of seedlings is an important indicator that reflects the size of the wheat population during the seedling stage. Researchers increasingly use deep learning to detect and count wheat seedlings from unmanned aerial vehicle (UAV) images. However, due to the small size and diverse postures of wheat seedlings, it can be challenging to estimate their numbers accurately during the seedling stage. In most related works in wheat seedling detection, they label the whole plant, often resulting in a higher proportion of soil background within the annotated bounding boxes. This imbalance between wheat seedlings and soil background in the annotated bounding boxes decreases the detection performance. This study proposes a wheat seedling detection method based on a local annotation instead of a global annotation. Moreover, the detection model is also improved by replacing convolutional and pooling layers with the Space-to-depth Conv module and adding a micro-scale detection layer in the YOLOv5 head network to better extract small-scale features in these small annotation boxes. The optimization of the detection model can reduce the number of error detections caused by leaf occlusion between wheat seedlings and the small size of wheat seedlings. The results show that the proposed method achieves a detection accuracy of 90.1%, outperforming other state-of-the-art detection methods. The proposed method provides a reference for future wheat seedling detection and yield prediction.
麦苗数量是反映苗期小麦群体大小的重要指标。研究人员越来越多地利用深度学习从无人机(UAV)图像中检测和计数麦苗。然而,由于麦苗个体较小且姿态多样,在苗期准确估计其数量具有挑战性。在大多数麦苗检测的相关工作中,他们对整株植物进行标注,这往往导致标注的边界框内土壤背景比例较高。标注边界框中麦苗与土壤背景之间的这种不平衡降低了检测性能。本研究提出了一种基于局部标注而非全局标注的麦苗检测方法。此外,通过用空间到深度卷积模块替换卷积层和池化层,并在YOLOv5头部网络中添加一个微尺度检测层,以更好地提取这些小标注框中的小尺度特征,对检测模型进行了改进。检测模型的优化可以减少麦苗之间叶片遮挡和麦苗个体较小导致的误检数量。结果表明,所提方法的检测准确率达到90.1%,优于其他先进的检测方法。所提方法为未来的麦苗检测和产量预测提供了参考。