Abstract
In order to improve the accuracy of apple detection, this essay proposes a detection method based on improved lightweight YOLOv5. By replacing the Depth Separable Convolution, the YOLOv5 algorithm goes through a lightweight modification. A visual attention mechanism model is additionally proposed, which is embedded in YOLOv5 to solve non-attention preference and parameter redundancy when extracting features in the network. The detection accuracy is therefore increased, and the computational burden brought by network parameters is reduced. Compared with the original algorithm, the detection speed can be increased optimally by 13.51%, and the average mAP can reach 95.03%. This method can basically fulfill the apple fruit identification on trees in natural environments.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
China Fruit Marking Association. Apple Market Forecast and Operational Recommendations. Hebei Science and Technology News, 13 October 2020 (003) (2020)
Zhang, J.: Empirical Analysis of Regional Differences of Apple’s Planting Cost and Benefit in China. Inner Mongolia Agricultural University (2018)
Papageorgiou, E.I., Aggelopoulou, K.D., Gemtos, T.A., et al.: Yield prediction in apples using fuzzy cognitive map learning approach. Comput. Electron. Agric. 91, 19–29 (2013)
Yang, X.H.: Research on visual navigation path method of kiwi fruit picking robot. Technol. Wind 15, 23–24 (2021)
Wan, J.Q., Fan, S.Z., Gong, L., et al.: Research status and development direction of design and control technology of fruit and vegetable picking robot system. Smart Agricult. 2(04), 17–40 (2020)
Zhao, D., Shen, T., Chen, Y., et al.: Fast tracking and recognition of overlapping fruit for apple harvesting robot. Trans. Chin. Soc. Agricult. Eng. 31(2), 22–28 (2015)
Fan, Y.Y., Zhang, Z.M., Chen, G.P., et al.: Application of vision sensor in the target fruit recognition system of picking robot. J. Agricult. Mechaniz. Res. 41(05), 210–214 (2019)
Feng, W., Zhao, D.A.: Fast tracking and recognition of target fruit of apple picking robot. Inf. Technol. 05, 5–9 (2018)
Su, M., Han, H.M.: Research on new equipment for fruit picking controlled by robot and manual assistance. Pract. Electron. Z1, 86–87 (2020)
Redmon, J., Divvala, S., Girshick, R., et al.: You only look once: unified, real-time object detection. In: The IEEE Conference on Computer Vision and Pattern Recognition, pp. 779–788. IEEE, Piscataway (2016)
Redmon, J., Farhadi, A.: YOLO9000: better, faster, stronger. In: The IEEE Conference on Computer Vision and Pattern Recognition, pp. 7263–7271. IEEE, Piscataway (2017)
Redmon, J., Farhadi, A.: YOLO v3: an incremental improvement [EB/OL] (2018). https://arxiv.org/abs/1804.02767
Bochkovskiy, A., Wang, C., Liao, H.: YOLOv4: optimal speed and accuracy of object detection [EB/OL] (2020). https://arxiv.org/abs/2004.10934
Liu, W., Anguelov, D., Erhan, D., et al.: SSD: single shot multibox detector. In: European Conference on Computer Vision, pp. 21–37. Springer, Cham (2016)
Zhang, S., Wen, L., Bian, X., et al.: Single-shot refinement neural network for object detection. The IEEE Conference on Computer Vision and Pattern Recognition, pp. 4203–4212. IEEE, Piscataway (2018)
Wang, D., Zhang, B., Cao, Y., et al.: SFSSD: shallow feature fusion single shot multibox detector. In: International Conference in Communications, Signal Processing, and Systems, pp. 2590–2598. Springer, Singapore (2019)
Yan, J.W., Zhao, Y., Zhang, L.Y., et al.: Recognition of prickly pear fruit in natural environment based on residual network. J. Chin. Agricult. Mechaniz. 41(10), 191–196 (2020)
Girshick, R., Donahue, J., Darrell, T., et al.: Rich feature hierarchies for accurate object detection and semantic segmentation. In: The IEEE Conference on Computer Vision and Pattern Recognition, pp. 580–587. IEEE, Piscataway (2014)
Girshick, R.: Fast R-CNN. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 1440–1448. IEEE, Piscataway (2015)
Ren, S., He, K., Girshick, R., et al.: Faster R-CNN: towards real-time object detection with region proposal networks. IEEE Trans. Pattern Anal. Mach. Intell. 39(6), 1137–1149 (2017)
Jing, W.B., Hu, H.T., Cheng, C., et al.: Ground apple recognition and counting based on deep learning. Jiangsu Agricult. Sci. 48(05), 210–219 (2020)
Zhou, W.H., Zhu, S.L.: Research on the application of smart examination room solutions based on deep learning technology. Construct. Budget Inf. Technol. Informatiz. 12, 224–227 (2020)
Wang, F.: Artificial intelligence detection and recognition algorithm for masks and helmets based on improved YOLOv5. Construct. Budget 11, 67–69 (2020)
Wang, C.Y., Liao, H.Y.M., Yeh, I.H., et al.: CSPNet: a new backbone that can enhance learning capability of CNN. In: The IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops, pp. 390–391. CVPRW, Seattle (2020)
Lin, T.Y., Dollar, P., Girshick, R., et al.: Feature pyramid networks for object detection. In: The IEEE Conference on Computer Vision and Pattern Recognition, pp. 2117–2125. IEEE, Piscataway (2017)
Liu, S., Qi, L., Qin, H., et al.: Path aggregation network for instance segmentation. In: The IEEE Conference on Computer Vision and Pattern Recognition, pp. 8759–8768. IEEE, Piscataway (2018)
Sifre, L., Mallat, S.: Rigid-motion scattering for texture classification. Comput. Sci. 3559, 501–515 (2014)
Shu, L., Zhang, Z.J., Lei, B.: Research on Dense-Yolov5 algorithm for infrared target detection. Opt. Optoelectron. Technol. 19(01), 69–75 (2021)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2022 Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Li, Z. et al. (2022). Detection Method of Apple Based on Improved Lightweight YOLOv5. In: Sun, F., Hu, D., Wermter, S., Yang, L., Liu, H., Fang, B. (eds) Cognitive Systems and Information Processing. ICCSIP 2021. Communications in Computer and Information Science, vol 1515. Springer, Singapore. https://doi.org/10.1007/978-981-16-9247-5_22
Download citation
DOI: https://doi.org/10.1007/978-981-16-9247-5_22
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-16-9246-8
Online ISBN: 978-981-16-9247-5
eBook Packages: Computer ScienceComputer Science (R0)