Development of a Lightweight Model for Rice Plant Counting and Localization Using UAV-Captured RGB Imagery

Abstract

Accurately obtaining both the number and the location of rice plants plays a critical role in agricultural applications, such as precision fertilization and yield prediction. With the rapid development of deep learning, numerous models for plant counting have been proposed. However, many of these models contain a large number of parameters, making them unsuitable for deployment in agricultural settings with limited computational resources. To address this challenge, we propose a novel pruning method, Cosine Norm Fusion (CNF), and a lightweight feature fusion technique, the Depth Attention Fusion Module (DAFM). Based on these innovations, we modify the existing P2PNet network to create P2P-CNF, a lightweight model for rice plant counting. The process begins with pruning the trained network using CNF, followed by the integration of our lightweight feature fusion module, DAFM. To validate the effectiveness of our method, we conducted experiments using rice datasets, including the RSC-UAV dataset, captured by UAV. The results demonstrate that our method achieves a MAE of 3.12 and an RMSE of 4.12 while utilizing only 33% of the original network parameters. We also evaluated our method on other plant counting datasets, and the results show that our method achieves a high counting accuracy while maintaining a lightweight architecture.

Code and data availability

The paper's core phenotyping asset is the authors' own RSC-UAV dataset (401 UAV RGB images with 69,768 annotated rice plant center points), which the Data Availability Statement says is available only on request from the corresponding author. The other datasets used (URC, WED, MTC) are cited prior-work datasets, not a

No evidence-backed public reproduction asset is currently recorded.