Pear Defect Detection Method Based on ResNet and DCGAN

被引:35
作者
Zhang, Yan [1 ]
Wa, Shiyun [1 ]
Sun, Pengshuo [1 ]
Wang, Yaojun [1 ]
机构
[1] China Agr Univ, Coll Informat & Elect Engn, Beijing 100083, Peoples R China
关键词
pear; defect detection; ResNet; deep convolutional generative adversarial network; convolutional neural network; iOS development; COMPUTER VISION; AGRICULTURE; DILATION; EROSION;
D O I
10.3390/info12100397
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
To address the current situation, in which pear defect detection is still based on a workforce with low efficiency, we propose the use of the CNN model to detect pear defects. Since it is challenging to obtain defect images in the implementation process, a deep convolutional adversarial generation network was used to augment the defect images. As the experimental results indicated, the detection accuracy of the proposed method on the 3000 validation set was as high as 97.35%. Variant mainstream CNNs were compared to evaluate the model's performance thoroughly, and the top performer was selected to conduct further comparative experiments with traditional machine learning methods, such as support vector machine algorithm, random forest algorithm, and k-nearest neighbor clustering algorithm. Moreover, the other two varieties of pears that have not been trained were chosen to validate the robustness and generalization capability of the model. The validation results illustrated that the proposed method is more accurate than the commonly used algorithms for pear defect detection. It is robust enough to be generalized well to other datasets. In order to allow the method proposed in this paper to be applied in agriculture, an intelligent pear defect detection system was built based on an iOS device.</p>
引用
收藏
页数:15
相关论文
共 34 条
[1]  
[Anonymous], 2017, P 2017 7 INT C INN C
[2]  
[Anonymous], 2021, IEEE Trans. Broadcast.
[3]  
Blum A., 1999, Proceedings of the Twelfth Annual Conference on Computational Learning Theory, P203, DOI 10.1145/307400.307439
[4]  
Bottou Leon, 2012, Neural Networks: Tricks of the Trade. Second Edition: LNCS 7700, P421, DOI 10.1007/978-3-642-35289-8_25
[5]   RECURSIVE EROSION, DILATION, OPENING, AND CLOSING TRANSFORMS [J].
CHEN, S ;
HARALICK, RM .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 1995, 4 (03) :335-345
[6]   Visual Detection Study on Early Bruises of Korla Pear Based on Hyperspectral Imaging Technology [J].
Chen Xin-xin ;
Guo Chen-tong ;
Zhang Chu ;
Liu Zi-yi ;
Jiang Hao ;
Lou Bing-gan ;
He Yong .
SPECTROSCOPY AND SPECTRAL ANALYSIS, 2017, 37 (01) :150-155
[7]   APPLICATION OF NEURAL NETWORKS IN IMAGE PROCESSING AND VISUALIZATION [J].
Cristea, Paul Dan .
GEOSPATIAL VISUAL ANALYTICS: GEOGRAPHICAL INFORMATION PROCESSING AND VISUAL ANALYTICS FOR ENVIRONMENTAL SECURITY, 2009, :59-71
[8]  
Cutkosky A., 2020, INT C MACHINE LEARNI, P2260
[9]  
El-Sawy A., 2016, CNN HANDWRITTEN ARAB
[10]  
Ge Z., 2021, 210708430 ARXIV