Enhancing cotton whitefly (Bemisia tabaci) detection and counting with a cost-effective deep learning approach on the Raspberry Pi

Abstract

Background The cotton whitefly (Bemisia tabaci) is a major global pest, causing significant crop damage through viral infestation and feeding. Traditional B. tabaci recognition relies on human eyes, which requires a large amount of work and high labor costs. The pests overlapping generations, high reproductive capacity, small size, and migratory behavior present challenges for the real-time monitoring and early warning systems. This study aims to develop an efficient, high-throughput automated system for detection of the cotton whiteflies. In this work, a novel tool for cotton whitefly fast identification and quantification was developed based on deep learning-based model. This approach enhances the effectiveness of B. tabaci control by facilitating earlier detection of its establishment in cotton, thereby allowing for a quicker implementation of management strategies. Results We compiled a dataset of 1200 annotated images of whiteflies on cotton leaves, augmented using techniques like flipping and rotation. We modified the YOLO v8s model by replacing the C2f module with the Swin-Transformer and introducing a P2 structure in the Head, achieving a precision of 0.87, mAP50 of 0.92, and F1 score of 0.88 through ablation studies. Additionally, we employed SAHI for image preprocessing and integrated the whitefly detection algorithm on a Raspberry Pi, and developed a GUI-based visual interface. Our preliminary analysis revealed a higher density of whiteflies on cotton leaves in the afternoon and the middle-top, middle, and middle-down plant sections. Conclusion Utilizing the enhanced YOLO v8s deep learning model, we have achieved precise detection and counting of whiteflies, enabling its application on hardware devices like the Raspberry Pi. This approach is highly suitable for research requiring accurate quantification of cotton whiteflies, including phenotypic analyses. Future work will focus on deploying such equipment in large fields to manage whitefly infestations.