Affiliation:
1. College of Engineering, Jiangxi Agricultural University, Nanchang 330045, China
2. College of Land Resources and Environment, Jiangxi Agricultural University, Nanchang 330045, China
Abstract
At present, Camellia oleifera fruit harvesting relies on manual labor with low efficiency, while mechanized harvesting could result in bud damage because flowering and fruiting are synchronized. As a prerequisite, rapid detection and identification are urgently needed for high accuracy and efficiency with simple models to realize selective and intelligent harvesting. In this paper, a lightweight detection algorithm YOLOv5s-Camellia based on YOLOv5s is proposed. First, the network unit of the lightweight network ShuffleNetv2 was used to reconstruct the backbone network, and thereby the number of computations and parameters of the model was reduced to increase the running speed for saving computational costs. Second, to mitigate the impact of the lightweight improvement on model detection accuracy, three efficient channel attention (ECA) modules were introduced into the backbone network to enhance the network’s attention to fruit features, and the Concat operation in the neck network was replaced by the Add operation with fewer parameters, which could increase the amount of information under features while maintaining the same number of channels. Third, the Gaussian Error Linear Units (GELU) activation function was introduced to improve the nonlinear characterization ability of the network. In addition, to improve the ability of the network to locate objects in the natural environment, the penalty index was redefined to optimize the bounding box loss function, which can improve the convergence speed and regression accuracy. Furthermore, the final experimental results showed that this model possesses 98.8% accuracy, 5.5 G FLOPs computation, and 6.3 MB size, and the detection speed reached 60.98 frame/s. Compared with the original algorithm, the calculation amount, size, and parameters were reduced by 65.18%, 56.55%, and 57.59%, respectively. The results can provide a technical reference for the development of a Camellia oleifera fruit-harvesting robot.
Funder
National Natural Science Foundation of China
National Key Research and Development Program of China
Jiangxi Provincial Forestry Bureau Camellia oleifera Fruit Research Special Project
Subject
Fluid Flow and Transfer Processes,Computer Science Applications,Process Chemistry and Technology,General Engineering,Instrumentation,General Materials Science
Reference39 articles.
1. Design and Experiment of Camellia oleifera Fruit Layered Harvesting Device;Rao;Trans. Chin. Soc. Agric Mac.,2021
2. Intelligent Method for Detection of Coronary Artery Disease with Ensemble Approach;Sapra;J. Adv. Commun. Comput. Technol.,2021
3. Intrusion Detection Systems in Cloud Computing Paradigm: Analysis and Overview;Rana;Complexity,2022
4. Verma, V., Gupta, D., Gupta, S., Uppal, M., Anand, D., Ortega-Mansilla, A., Alharithi, F.S., Almotiri, J., and Goyal, N. (2022). A Deep Learning-Based Intelligent Garbage Detection System Using an Unmanned Aerial Vehicle. Symmetry., 14.
5. Fruit detection and positioning technology for a Camellia oleifera C. Abel orchard based on improved YOLOv4-tiny model and binocular stereo vision;Tang;J. Expert Syst. Appl.,2023
Cited by
2 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献