Title: A study of an improved RCNN network model for surface defect detection algorithm of precision workpiece and its realisation

Authors: Xiaohong Dai; Yingji Zhao; Chaoping Zhu

Addresses: Chongqing Key Laboratory of E-commerce and Supply Chain System, Chongqing Technology and Business University, Chongqing 400067, China ' School of Foreign Languages, Chongqing University of Science and Technology, Chongqing 401331, China ' School of Artificial Intelligence, Chongqing Technology and Business University, Chongqing 400067, China

Abstract: In this paper, an improved RCNN algorithm for surface defect detection of precision workpiece is proposed. During model training, the original image is processed by data amplification, which overcomes the phenomenon of model overfitting caused by too small data set. The fuzzy image is processed by the conditional generation antagonism network, which guarantees the authenticity of the image and eliminates the fuzziness of the image. On the basis of network selection, through network training and comparative analysis, ResNet101 was selected as the basic network for feature extraction. By combining the network performance, the number of convolutional layers (model depth), system efficiency and other factors, and taking advantage of the spatial characteristics of images, the irregular cross convolution kernel idea and the differentiated convolution kernel design method are adopted to realise the feature fusion of different convolutional layers. Embedded in the target Network Squeeze and Excitation (SE) module channel features fusion module, the Feature Pyramid Network (FPN) module, ROI Network module. The experimental results show that the improved network model is used to realise the location and classification of defects on the workpiece surface. The detection accuracy was 91% and the recall rate was 93%.

Keywords: precision workpiece; defect identification; convolution neural network; deep learning; target detection.

DOI: 10.1504/IJWMC.2020.109275

International Journal of Wireless and Mobile Computing, 2020 Vol.19 No.1, pp.95 - 105

Received: 28 Mar 2020
Accepted: 15 May 2020
Published online: 02 Sep 2020 *

Full-text access for editors Full-text access for subscribers Purchase this article Comment on this article