SoybeanNet: Transformer-Based Convolutional Neural Network for Soybean Pod Counting from Unmanned Aerial Vehicle (UAV) Images

Jiajia Li,Raju Thada Magar,Dong Chen,Feng Lin,Dechun Wang,Xiang Yin,Weichao Zhuang,Zhaojian Li
2023-11-19
Abstract:Soybeans are a critical source of food, protein and oil, and thus have received extensive research aimed at enhancing their yield, refining cultivation practices, and advancing soybean breeding techniques. Within this context, soybean pod counting plays an essential role in understanding and optimizing production. Despite recent advancements, the development of a robust pod-counting algorithm capable of performing effectively in real-field conditions remains a significant challenge This paper presents a pioneering work of accurate soybean pod counting utilizing unmanned aerial vehicle (UAV) images captured from actual soybean fields in Michigan, USA. Specifically, this paper presents SoybeanNet, a novel point-based counting network that harnesses powerful transformer backbones for simultaneous soybean pod counting and localization with high accuracy. In addition, a new dataset of UAV-acquired images for soybean pod counting was created and open-sourced, consisting of 113 drone images with more than 260k manually annotated soybean pods captured under natural lighting conditions. Through comprehensive evaluations, SoybeanNet demonstrated superior performance over five state-of-the-art approaches when tested on the collected images. Remarkably, SoybeanNet achieved a counting accuracy of $84.51\%$ when tested on the testing dataset, attesting to its efficacy in real-world scenarios. The publication also provides both the source code (\url{<a class="link-external link-https" href="https://github.com/JiajiaLi04/Soybean-Pod-Counting-from-UAV-Images" rel="external noopener nofollow">this https URL</a>}) and the labeled soybean dataset (\url{<a class="link-external link-https" href="https://www.kaggle.com/datasets/jiajiali/uav-based-soybean-pod-images" rel="external noopener nofollow">this https URL</a>}), offering a valuable resource for future research endeavors in soybean pod counting and related fields.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The paper primarily addresses the issue of soybean pod counting. Specifically: - **Research Background and Objectives**: Soybeans, as an important source of food, protein, and oil, hold significant economic importance globally. To meet the growing population demand and increase yield, researchers are dedicated to developing accurate soybean pod counting methods, which are crucial for optimizing production, improving breeding techniques, and enhancing genetic gains. - **Existing Challenges**: Although previous studies have attempted plant counting using machine learning and deep learning techniques, soybean pod counting still faces numerous challenges, including small pod size, high density, severe overlap, and especially under natural field conditions. - **Contributions of the Paper**: - Proposed a soybean pod point annotation dataset based on UAV image acquisition, containing 113 images and over 260,000 manually annotated soybean pods. These images were obtained from real field environments under natural lighting conditions. - Developed a novel point-based counting network named SoybeanNet, which utilizes a powerful Transformer backbone to achieve simultaneous counting and localization of soybean pods with high accuracy. - Experimental results show that SoybeanNet achieved a counting accuracy of 84.51% on the test dataset, significantly outperforming five state-of-the-art methods, demonstrating its effectiveness in real-world agricultural environments. - **Method Overview**: The paper introduces a method based on the Transformer architecture to address the soybean pod counting problem. First, UAV images from real fields were collected and carefully point-annotated. Subsequently, the SoybeanNet framework was proposed, consisting of a Transformer backbone, regression branch, classification branch, and a one-to-one matcher based on the Hungarian algorithm. The effectiveness of this method was evaluated using a series of metrics (such as Mean Absolute Error MAE, Root Mean Square Error RMSE, etc.) and compared with other advanced methods. In summary, this paper aims to develop an efficient and accurate soybean pod counting method to support breeding work and agricultural production decision-making. By introducing a new dataset and advanced deep learning models, it provides a powerful support tool for soybean pod counting.