您好,欢迎访问北京市农林科学院 机构知识库!

Using filter pruning-based deep learning algorithm for the real-time fruit freshness detection with edge processors

文献类型: 外文期刊

作者: Mao, DianHui 1 ; Zhang, DengHui 1 ; Sun, Hao 1 ; Wu, JianWei 3 ; Chen, JunHua 5 ;

作者机构: 1.Beijing Technol & Business Univ, Beijing Key Lab Big Data Technol Food Safety, Sch Comp Sci & Engn, Beijing 100048, Peoples R China

2.Beijing Technol & Business Univ, Natl Engn Lab Agriprod Qual Traceabil, Beijing 100048, Peoples R China

3.Beijing Acad Agr & Forestry Sci, Informat Technol Res Ctr, Beijing 100048, Peoples R China

4.Beijing PAIDE Sci & Technol Dev Co Ltd, Beijing 100097, Peoples R China

5.China Natl Inst Standardizat, Standardizat Theory & Strategy, Beijing 100088, Peoples R China

关键词: PP-YOLO Tiny; Ultra Lightweight; FPGM algorithm; Real-time detection; Fruit

期刊名称:JOURNAL OF FOOD MEASUREMENT AND CHARACTERIZATION ( 影响因子:3.4; 五年影响因子:3.2 )

ISSN: 2193-4126

年卷期: 2024 年 18 卷 2 期

页码:

收录情况: SCI

摘要: In the context of edge devices used in horticulture and the fruit industry, traditional high-precision models prove to be computationally intensive and challenging to deploy on terminals with limited computing resources. Therefore, in this work, we propose a lightweight target detection algorithm, called Fruit-Yolo, applied to edge farming devices where there are insufficient hardware resources and weak computational power to achieve classification and localization of fruit. More specifically, we employ the PPYOLO-Tiny model as our benchmark, where we undertake the redesign of both the backbone and FPN network components of the model. Additionally, we adapt the anchor framework to better suit the requirements of fruit detection, ultimately resulting in the creation of our streamlined network (Fruit-Yolo). Then, the filtered pruning based on geometric median algorithm is used to prune the model. The test results showed that the number of parameters of model after pruning was reduced by 20%, and the mAP decreased by only 0.2%. To verify the effectiveness of the proposed method, five algorithms including theYOLOv5s, SSD 300, MobileNetv3 and Faster R-CNN were compared. The comparison results show that the Fruit-Yolo model has the smallest size (3.85 MB), high accuracy (98.6%) and fast detection speed (32.77FPS). The results indicated that the proposed method can provide a technical reference for the deployment of deep learning models in intelligent agricultural machines at the edge.

  • 相关文献
作者其他论文 更多>>