Lightweight Fruit-Detection Algorithm for Edge Computing Applications

In recent years, deep-learning-based fruit-detection technology has exhibited excellent performance in modern horticulture research. However, deploying deep learning algorithms in real-time field applications is still challenging, owing to the relatively low image processing capability of edge devic...

Full description

Bibliographic Details
Main Authors: Wenli Zhang, Yuxin Liu, Kaizhen Chen, Huibin Li, Yulin Duan, Wenbin Wu, Yun Shi, Wei Guo
Format: Article
Language:English
Published: Frontiers Media S.A. 2021-10-01
Series:Frontiers in Plant Science
Subjects:
Online Access:https://www.frontiersin.org/articles/10.3389/fpls.2021.740936/full
_version_ 1818989373163569152
author Wenli Zhang
Yuxin Liu
Kaizhen Chen
Huibin Li
Huibin Li
Yulin Duan
Yulin Duan
Wenbin Wu
Wenbin Wu
Yun Shi
Yun Shi
Wei Guo
author_facet Wenli Zhang
Yuxin Liu
Kaizhen Chen
Huibin Li
Huibin Li
Yulin Duan
Yulin Duan
Wenbin Wu
Wenbin Wu
Yun Shi
Yun Shi
Wei Guo
author_sort Wenli Zhang
collection DOAJ
description In recent years, deep-learning-based fruit-detection technology has exhibited excellent performance in modern horticulture research. However, deploying deep learning algorithms in real-time field applications is still challenging, owing to the relatively low image processing capability of edge devices. Such limitations are becoming a new bottleneck and hindering the utilization of AI algorithms in modern horticulture. In this paper, we propose a lightweight fruit-detection algorithm, specifically designed for edge devices. The algorithm is based on Light-CSPNet as the backbone network, an improved feature-extraction module, a down-sampling method, and a feature-fusion module, and it ensures real-time detection on edge devices while maintaining the fruit-detection accuracy. The proposed algorithm was tested on three edge devices: NVIDIA Jetson Xavier NX, NVIDIA Jetson TX2, and NVIDIA Jetson NANO. The experimental results show that the average detection precision of the proposed algorithm for orange, tomato, and apple datasets are 0.93, 0.847, and 0.850, respectively. Deploying the algorithm, the detection speed of NVIDIA Jetson Xavier NX reaches 21.3, 24.8, and 22.2 FPS, while that of NVIDIA Jetson TX2 reaches 13.9, 14.1, and 14.5 FPS and that of NVIDIA Jetson NANO reaches 6.3, 5.0, and 8.5 FPS for the three datasets. Additionally, the proposed algorithm provides a component add/remove function to flexibly adjust the model structure, considering the trade-off between the detection accuracy and speed in practical usage.
first_indexed 2024-12-20T19:37:26Z
format Article
id doaj.art-8a0628612a8e49a5919662efbb66e5ad
institution Directory Open Access Journal
issn 1664-462X
language English
last_indexed 2024-12-20T19:37:26Z
publishDate 2021-10-01
publisher Frontiers Media S.A.
record_format Article
series Frontiers in Plant Science
spelling doaj.art-8a0628612a8e49a5919662efbb66e5ad2022-12-21T19:28:37ZengFrontiers Media S.A.Frontiers in Plant Science1664-462X2021-10-011210.3389/fpls.2021.740936740936Lightweight Fruit-Detection Algorithm for Edge Computing ApplicationsWenli Zhang0Yuxin Liu1Kaizhen Chen2Huibin Li3Huibin Li4Yulin Duan5Yulin Duan6Wenbin Wu7Wenbin Wu8Yun Shi9Yun Shi10Wei Guo11Department of Information, Beijing University of Technology, Beijing, ChinaDepartment of Information, Beijing University of Technology, Beijing, ChinaDepartment of Information, Beijing University of Technology, Beijing, ChinaInstitute of Agricultural Resources and Regional Planning, Chinese Academy of Agricultural Sciences, Beijing, ChinaKey Laboratory of Agricultural Remote Sensing, Ministry of Agriculture, Beijing, ChinaInstitute of Agricultural Resources and Regional Planning, Chinese Academy of Agricultural Sciences, Beijing, ChinaKey Laboratory of Agricultural Remote Sensing, Ministry of Agriculture, Beijing, ChinaInstitute of Agricultural Resources and Regional Planning, Chinese Academy of Agricultural Sciences, Beijing, ChinaKey Laboratory of Agricultural Remote Sensing, Ministry of Agriculture, Beijing, ChinaInstitute of Agricultural Resources and Regional Planning, Chinese Academy of Agricultural Sciences, Beijing, ChinaKey Laboratory of Agricultural Remote Sensing, Ministry of Agriculture, Beijing, ChinaInternational Field Phenomics Research Laboratory, Institute for Sustainable Agro-Ecosystem Services, The University of Tokyo, Tokyo, JapanIn recent years, deep-learning-based fruit-detection technology has exhibited excellent performance in modern horticulture research. However, deploying deep learning algorithms in real-time field applications is still challenging, owing to the relatively low image processing capability of edge devices. Such limitations are becoming a new bottleneck and hindering the utilization of AI algorithms in modern horticulture. In this paper, we propose a lightweight fruit-detection algorithm, specifically designed for edge devices. The algorithm is based on Light-CSPNet as the backbone network, an improved feature-extraction module, a down-sampling method, and a feature-fusion module, and it ensures real-time detection on edge devices while maintaining the fruit-detection accuracy. The proposed algorithm was tested on three edge devices: NVIDIA Jetson Xavier NX, NVIDIA Jetson TX2, and NVIDIA Jetson NANO. The experimental results show that the average detection precision of the proposed algorithm for orange, tomato, and apple datasets are 0.93, 0.847, and 0.850, respectively. Deploying the algorithm, the detection speed of NVIDIA Jetson Xavier NX reaches 21.3, 24.8, and 22.2 FPS, while that of NVIDIA Jetson TX2 reaches 13.9, 14.1, and 14.5 FPS and that of NVIDIA Jetson NANO reaches 6.3, 5.0, and 8.5 FPS for the three datasets. Additionally, the proposed algorithm provides a component add/remove function to flexibly adjust the model structure, considering the trade-off between the detection accuracy and speed in practical usage.https://www.frontiersin.org/articles/10.3389/fpls.2021.740936/fullmodern horticulturedeep learningfruit detectionlightweightedge devices
spellingShingle Wenli Zhang
Yuxin Liu
Kaizhen Chen
Huibin Li
Huibin Li
Yulin Duan
Yulin Duan
Wenbin Wu
Wenbin Wu
Yun Shi
Yun Shi
Wei Guo
Lightweight Fruit-Detection Algorithm for Edge Computing Applications
Frontiers in Plant Science
modern horticulture
deep learning
fruit detection
lightweight
edge devices
title Lightweight Fruit-Detection Algorithm for Edge Computing Applications
title_full Lightweight Fruit-Detection Algorithm for Edge Computing Applications
title_fullStr Lightweight Fruit-Detection Algorithm for Edge Computing Applications
title_full_unstemmed Lightweight Fruit-Detection Algorithm for Edge Computing Applications
title_short Lightweight Fruit-Detection Algorithm for Edge Computing Applications
title_sort lightweight fruit detection algorithm for edge computing applications
topic modern horticulture
deep learning
fruit detection
lightweight
edge devices
url https://www.frontiersin.org/articles/10.3389/fpls.2021.740936/full
work_keys_str_mv AT wenlizhang lightweightfruitdetectionalgorithmforedgecomputingapplications
AT yuxinliu lightweightfruitdetectionalgorithmforedgecomputingapplications
AT kaizhenchen lightweightfruitdetectionalgorithmforedgecomputingapplications
AT huibinli lightweightfruitdetectionalgorithmforedgecomputingapplications
AT huibinli lightweightfruitdetectionalgorithmforedgecomputingapplications
AT yulinduan lightweightfruitdetectionalgorithmforedgecomputingapplications
AT yulinduan lightweightfruitdetectionalgorithmforedgecomputingapplications
AT wenbinwu lightweightfruitdetectionalgorithmforedgecomputingapplications
AT wenbinwu lightweightfruitdetectionalgorithmforedgecomputingapplications
AT yunshi lightweightfruitdetectionalgorithmforedgecomputingapplications
AT yunshi lightweightfruitdetectionalgorithmforedgecomputingapplications
AT weiguo lightweightfruitdetectionalgorithmforedgecomputingapplications