Fast and accurate detection of kiwifruit in orchard using improved YOLOv3-tiny model

被引:163
作者
Fu, Longsheng [1 ,3 ,4 ,5 ]
Feng, Yali [1 ,6 ]
Wu, Jingzhu [2 ]
Liu, Zhihao [1 ]
Gao, Fangfang [1 ]
Majeed, Yaqoob [5 ]
Al-Mallahi, Ahmad [7 ]
Zhang, Qin [5 ]
Li, Rui [1 ]
Cui, Yongjie [1 ]
机构
[1] Northwest A&F Univ, Coll Mech & Elect Engn, Yangling 712100, Shaanxi, Peoples R China
[2] Beijing Technol & Business Univ, Beijing Key Lab Big Data Technol Food Safety, Beijing 100048, Peoples R China
[3] Minist Agr & Rural Affairs, Key Lab Agr Internet Things, Yangling 712100, Shaanxi, Peoples R China
[4] Shaanxi Key Lab Agr Informat Percept & Intelligen, Yangling 712100, Shaanxi, Peoples R China
[5] Washington State Univ, Ctr Precis & Automated Agr Syst, Prosser, WA 99350 USA
[6] Shanxi Agr Univ, Coll Engn, Jinzhong 030801, Shaanxi, Peoples R China
[7] Dalhousie Univ, Dept Engn, Fac Agr, Truro, NS B2N 5E3, Canada
关键词
Data augmentation; Image detection; Deep learning; YOLOv3-tiny model; Convolutional kernel; APPLE DETECTION; MACHINE VISION; R-CNN; RECOGNITION; CLASSIFICATION; IMAGES; FRUITS;
D O I
10.1007/s11119-020-09754-y
中图分类号
S [农业科学];
学科分类号
09 ;
摘要
Automatic detection of kiwifruit in the orchard is challenging because illumination varies through the day and night and because of color similarity between kiwifruit and the complex background of leaves, branches and stems. Also, kiwifruits grow in clusters, which may result in having occluded and touching fruits. A fast and accurate object detection algorithm was developed to automatically detect kiwifruits in the orchard by improving the YOLOv3-tiny model. Based on the characteristics of kiwifruit images, two convolutional kernels of 3 x 3 and 1 x 1 were added to the fifth and sixth convolution layers of the YOLOv3-tiny model, respectively, to develop a deep YOLOv3-tiny (DY3TNet) model. It takes multiple 1 x 1 convolutional layers in intermediate layers of the network to reduce the computational complexity. Testing images captured from day and night and comparing with other deep learning models, namely, Faster R-CNN with ZFNet, Faster R-CNN with VGG16, YOLOv2 and YOLOv3-tiny, the DY3TNet model achieved the highest average precision of 0.9005 with the smallest data weight of 27 MB. Furthermore, it took only 34 ms on average to process an image of a resolution of 2352 x 1568 pixels. The DY3TNet model, along with the YOLOv3-tiny model, showed better performance on images captured with flash than those without. Moreover, the experiments indicated that the image augmentation process could improve the detection performance, and a simple lighting arrangement could improve the success rate of detection in the orchard. The experimental results demonstrated that the improved DY3TNet model is small and efficient and that it would increase the applicability of real-time kiwifruit detection in the orchard even when small hardware devices are used.
引用
收藏
页码:754 / 776
页数:23
相关论文
共 59 条
[1]   Using Deep Convolutional Neural Network Architectures for Object Classification and Detection Within X-Ray Baggage Security Imagery [J].
Akcay, Samet ;
Kundegorski, Mikolaj E. ;
Willcocks, Chris G. ;
Breckon, Toby P. .
IEEE TRANSACTIONS ON INFORMATION FORENSICS AND SECURITY, 2018, 13 (09) :2203-2215
[2]   Simultaneous detection and classification of breast masses in digital mammograms via a deep learning YOLO-based CAD system [J].
Al-masni, Mohammed A. ;
Al-antari, Mugahed A. ;
Park, Jeong-Min ;
Gi, Geon ;
Kim, Tae-Yeon ;
Rivera, Patricio ;
Valarezo, Edwin ;
Choi, Mun-Taek ;
Han, Seung-Moo ;
Kim, Tae-Seong .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2018, 157 :85-94
[3]  
[Anonymous], 2017, THESIS
[4]   Image Segmentation for Fruit Detection and Yield Estimation in Apple Orchards [J].
Bargoti, Suchet ;
Underwood, James P. .
JOURNAL OF FIELD ROBOTICS, 2017, 34 (06) :1039-1060
[5]   Faster R-CNN-based apple detection in dense-foliage fruiting-wall trees using RGB and depth features for robotic harvesting [J].
Fu, Longsheng ;
Majeed, Yaqoob ;
Zhang, Xin ;
Karkee, Manoj ;
Zhang, Qin .
BIOSYSTEMS ENGINEERING, 2020, 197 :245-256
[6]   A novel image processing algorithm to separate linearly clustered kiwifruits [J].
Fu, Longsheng ;
Tola, Elkamil ;
Al-Mallahi, Ahmad ;
Li, Rui ;
Cui, Yongjie .
BIOSYSTEMS ENGINEERING, 2019, 183 :184-195
[7]   Kiwifruit detection in field images using Faster R-CNN with ZFNet [J].
Fu, Longsheng ;
Feng, Yali ;
Majeed, Yaqoob ;
Zhang, Xin ;
Zhang, Jing ;
Karkee, Manoj ;
Zhang, Qin .
IFAC PAPERSONLINE, 2018, 51 (17) :45-50
[8]   Preparation of Papers for IFAC Conferences & Symposia: Kiwifruit yield estimation using image processing by an Android mobile phone [J].
Fu, Longsheng ;
Liu, Zhihao ;
Majeed, Yaqoob ;
Cui, Yongjie .
IFAC PAPERSONLINE, 2018, 51 (17) :185-190
[9]   Classification of Kiwifruit Grades Based on Fruit Shape Using a Single Camera [J].
Fu, Longsheng ;
Sun, Shipeng ;
Li, Rui ;
Wang, Shaojin .
SENSORS, 2016, 16 (07)
[10]   Kiwifruit recognition at nighttime using artificial lighting based on machine vision [J].
Fu Longsheng ;
Wang Bin ;
Cui Yongjie ;
Su Shuai ;
Gejima, Yoshinori ;
Kobayashi, Taiichi .
INTERNATIONAL JOURNAL OF AGRICULTURAL AND BIOLOGICAL ENGINEERING, 2015, 8 (04) :52-59