A lightweight detection model for rice grain with dense bonding distribution based on YOLOv5s

Abstract

Accurate counting of rice grains plays a critical role in rice breeding and thousand-grain weight measurement. However, the accuracy of existing algorithms is insufficient under conditions of high-density and densely bonded rice grain distribution. To quickly and accurately detect high-density and densely bonded rice grains with as many grains as possible, this study developed a lightweight model based on YOLOv5s. First, we built an efficient lightweight model architecture to obtain small-target location and semantic information of rice grains. Second, we use an omni-dimensional dynamic convolution (ODConv) module to replace some of the convolutions of the backbone network to fully extract feature information. We then introduce the mixed local channel attention (MLCA) mechanism to weigh local features through spatial information, allowing the model to locate and identify dense rice grains accurately. Finally, we use the SIoU loss function to improve the convergence speed and accuracy of model training. The model’s detection accuracy was verified via ablation experiments. The results indicated that compared with the original YOLOv5s network, the model size, parameters and floating-point operations per second (FLOPS) of the improved model decreased by 64.16 %, 70.8 % and 28.3 %, respectively, while mAP₀.₅:₀.₉₅ increased by 7.21 %. The mean error rate and mean detection time of the improved model were 0.234 % and 25.9 ms, respectively. Its superior capacity against other detection algorithm models at rapidly detecting densely bonded rice grains. Furthermore, an android application was further developed. After comparative testing on three types of mobile phones, the application was able to effectively and accurately detect and count rice grains, providing an effective solution for rice grain detection and counting.

Plant phenotyping relevance

米粒の検出・計数という植物形質取得を目的に、YOLOv5s改良モデルを開発し、アブレーション実験と比較評価、モバイルアプリ実装まで行っており、フェノタイピング手法が研究の中心である。

Code and data availability

公開状態または取得可能な本文経路を確認できませんでした。

No evidence-backed public reproduction asset is currently recorded.