Lightweight Steel Surface Defect Detection Algorithm Based on Improved RetinaNet
For the requirement of the practical application,the existing defect detection algorithms suffer from the problems of slow detection speed and low detection accuracy.To address these issues,a lightweight steel surface defect detection algorithm based on improved RetinaNet is proposed.Firstly,the original backbone network is replaced by a lightweight network,and a cross-stage-partial structure is introduced to achieve effective propagation and lightweighting of gradients.Then,depth-separable convolution is employed to replace the traditional convolutional layer to further reduce the number of parameters and improve the detection speed.To compensate for the decrease in model accuracy caused by lightweighting,a spatial pyramid pooling mechanism based on the cross-stage partial structure is designed.The detection accuracy of the model is effectively improved by feature fusion at different scales.Finally,experiments on NEU-DET dataset and the self-built HBIS dataset demonstrate the proposed algorithm reaches a faster detection speed and higher accuracy.Moreover,the corresponding hardware and software system meets the real-time online detection requirements of the production line and it has been put into service.
万方数据
维普
