1. PcMNet: An efficient lightweight apple detection algorithm in natural orchards
- Author
-
Shiwei Wen, Jianguo Zhou, Guangrui Hu, Hao Zhang, Shan Tao, Zeyu Wang, and Jun Chen
- Subjects
Apple detection ,Lightweight model ,YOLOv8 ,Partial convolution ,Faster-CCFF ,Edge computing devices ,Agriculture (General) ,S1-972 ,Agricultural industries ,HD9000-9495 - Abstract
Apple detection plays a critical role in enabling the functionality of harvesting robots within natural orchard environments. To address challenges related to low detection accuracy, slow inference speed, and high parameter count, we present PcMNet, a lightweight detection model based on an improved YOLOv8 network. Initially, we employed Partial Convolution (Pconv) to construct a PR module, forming the Pconv-block, which subsequently replaced the original C2f feature extraction module within the YOLOv8n backbone. This replacement led to improvements in both detection accuracy and speed, while simultaneously reducing computational complexity (FLOPs), parameter count, and model size. Furthermore, the Cross-Scale Feature Fusion (CCFF) module was refined into Faster-Cross-Scale Feature Fusion (Faster-CCFF) with the integration of Pconv-block, significantly enhancing the model's feature extraction and fusion capabilities. Additionally, we introduced Mixed Local Channel Attention (MLCA) to further strengthen the model's capacity to capture essential features while effectively suppressing background noise. Experimental results demonstrate that PcMNet achieved a detection accuracy of 92.8 % and an mAP@0.5 of 95.5 %, representing improvements of 1.4 and 0.7 percentage points, respectively, over YOLOv8n. Moreover, PcMNet successfully reduced FLOPs, parameter count, and model size to 5.1 G, 1.4 M, and 3.2 MB, respectively. The per-image detection time was reduced to 2.3 ms, indicating reductions of 37.80 %, 53.33 %, 49.21 %, and 56.60 % in FLOPs, parameters, model size, and detection time compared to YOLOv8n. When deployed on edge computing devices with TensorRT acceleration, PcMNet achieved a detection rate of 92 FPS. Field validation experiments conducted in natural orchard environments confirmed PcMNet's superior ability to detect apples under challenging conditions, such as occlusions and varying lighting conditions. Its lightweight design and rapid detection capabilities provide a valuable reference for achieving real-time apple detection in automated and intelligent harvesting robots, thereby contributing to advancements in smart agriculture.
- Published
- 2024
- Full Text
- View/download PDF