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

文献类型： 外文期刊

第一作者： Mao, DianHui

作者： Mao, DianHui;Zhang, DengHui;Sun, Hao;Mao, DianHui;Wu, JianWei;Wu, JianWei;Chen, JunHua

作者机构：

关键词： 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.

分类号：

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

微信小程序