Abstract:Passion fruit, renowned for its significant nutritional, medicinal, and economic value, is extensively cultivated in subtropical regions such as China, India, and Vietnam. In the production and processing industry, the quality grading of passion fruit plays a crucial role in the supply chain. However, the current process relies heavily on manual labor, resulting in inefficiency and high costs, which reflects the importance of expanding the application of fruit appearance quality classification mechanisms based on computer vision. Moreover, the existing passion fruit detection algorithms mainly focus on real-time detection and overlook the quality-classification aspect. This paper proposes the ATC-YOLOv5 model based on deep learning for passion fruit detection and quality classification. First, an improved Asymptotic Feature Pyramid Network (APFN) is utilized as the feature-extraction network, which is the network modified in this study by adding weighted feature concat pathways. This optimization enhances the feature flow between different levels and nodes, allowing for the adaptive and asymptotic fusion of richer feature information related to passion fruit quality. Secondly, the Transformer Cross Stage Partial (TRCSP) layer is constructed based on the introduction of the Multi-Head Self-Attention (MHSA) layer in the Cross Stage Partial (CSP) layer, enabling the network to achieve a better performance in modeling long-range dependencies. In addition, the Coordinate Attention (CA) mechanism is introduced to enhance the network's learning capacity for both local and non-local information, as well as the fine-grained features of passion fruit. Moreover, to validate the performance of the proposed model, a self-made passion fruit dataset is constructed to classify passion fruit into four quality grades. The original YOLOv5 serves as the baseline model. According to the experimental results, the mean average precision (mAP) of ATC-YOLOv5 reaches 95.36%, and the mean detection time (mDT) is 3.2 ms, which improves the mAP by 4.83% and the detection speed by 11.1%, and the number of parameters is reduced by 10.54% compared to the baseline, maintaining the lightweight characteristics while improving the accuracy. These experimental results validate the high detection efficiency of the proposed model for fruit quality classification, contributing to the realization of intelligent agriculture and fruit industries.

A fast and lightweight detection algorithm for passion fruit pests based on improved YOLOv5

Research and Explainable Analysis of a Real-Time Passion Fruit Detection Model Based on FSOne-YOLOv7

A Lightweight and High-Precision Passion Fruit YOLO Detection Model for Deployment in Embedded Devices

Fast Detection of Passion Fruit with Multi-class Based on YOLOv3

ATC-YOLOv5: Fruit Appearance Quality Classification Algorithm Based on the Improved YOLOv5 Model for Passion Fruits

Real-time and effective detection of agricultural pest using an improved YOLOv5 network

AgriPest-YOLO: A rapid light-trap agricultural pest detection method based on deep learning

A real-time object detection model for orchard pests based on improved YOLOv4 algorithm

A Real-Time Kiwifruit Detection Based on Improved YOLOv7.

ED-YOLOv8s: an Enhanced Approach for Passion Fruit Maturity Detection Based on YOLOv8s

Pest-YOLO: A Lightweight Pest Detection Model Based on Multi-level Feature Fusion

Lightweight Detection Algorithm for Corn Insect Pests Based on Attention Mechanism

Pest-YOLO: A model for large-scale multi-class dense and tiny pest detection and counting

Rapid detection of Yunnan Xiaomila based on lightweight YOLOv7 algorithm

PestLite: A Novel YOLO-Based Deep Learning Technique for Crop Pest Detection

Lightweight Crop Pest Identification Algorithm under Natural Background

Tomato Pest Recognition Algorithm Based on Improved YOLOv4

MS-YOLOv5: a lightweight algorithm for strawberry ripeness detection based on deep learning

Real-time and Accurate Detection of Citrus in Complex Scenes Based on HPL-YOLOv4.

Maize-YOLO: A New High-Precision and Real-Time Method for Maize Pest Detection

YOLOv7‐PSAFP: Crop pest and disease detection based on improved YOLOv7