A Real-Time Apple Targets Detection Method for Picking Robot Based on ShufflenetV2-YOLOX
Wei Ji,
Yu Pan,
Bo Xu and
Juncheng Wang
Additional contact information
Wei Ji: School of Electrical and Information Engineering, Jiangsu University, Zhenjiang 212013, China
Yu Pan: School of Electrical and Information Engineering, Jiangsu University, Zhenjiang 212013, China
Bo Xu: School of Electrical and Information Engineering, Jiangsu University, Zhenjiang 212013, China
Juncheng Wang: School of Electrical and Information Engineering, Jiangsu University, Zhenjiang 212013, China
Agriculture, 2022, vol. 12, issue 6, 1-18
Abstract:
In order to enable the picking robot to detect and locate apples quickly and accurately in the orchard natural environment, we propose an apple object detection method based on Shufflenetv2-YOLOX. This method takes YOLOX-Tiny as the baseline and uses the lightweight network Shufflenetv2 added with the convolutional block attention module (CBAM) as the backbone. An adaptive spatial feature fusion (ASFF) module is added to the PANet network to improve the detection accuracy, and only two extraction layers are used to simplify the network structure. The average precision (AP), precision, recall, and F1 of the trained network under the verification set are 96.76%, 95.62%, 93.75%, and 0.95, respectively, and the detection speed reaches 65 frames per second (FPS). The test results show that the AP value of Shufflenetv2-YOLOX is increased by 6.24% compared with YOLOX-Tiny, and the detection speed is increased by 18%. At the same time, it has a better detection effect and speed than the advanced lightweight networks YOLOv5-s, Efficientdet-d0, YOLOv4-Tiny, and Mobilenet-YOLOv4-Lite. Meanwhile, the half-precision floating-point (FP16) accuracy model on the embedded device Jetson Nano with TensorRT acceleration can reach 26.3 FPS. This method can provide an effective solution for the vision system of the apple picking robot.
Keywords: machine vision; picking robot; apple detection; YOLOX; ShufflenetV2 (search for similar items in EconPapers)
JEL-codes: Q1 Q10 Q11 Q12 Q13 Q14 Q15 Q16 Q17 Q18 (search for similar items in EconPapers)
Date: 2022
References: View complete reference list from CitEc
Citations: View citations in EconPapers (2)
Downloads: (external link)
https://www.mdpi.com/2077-0472/12/6/856/pdf (application/pdf)
https://www.mdpi.com/2077-0472/12/6/856/ (text/html)
Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.
Export reference: BibTeX
RIS (EndNote, ProCite, RefMan)
HTML/Text
Persistent link: https://EconPapers.repec.org/RePEc:gam:jagris:v:12:y:2022:i:6:p:856-:d:837836
Access Statistics for this article
Agriculture is currently edited by Ms. Leda Xuan
More articles in Agriculture from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().