Wheel hub defect detection based on the DS-Cascade RCNN.

• Deformable convolution is used in the feature extraction network. The position and size of the convolution kernel are dynamically adjusted according to the shape of the wheel hub defects. • In this paper, global information is introduced from the spatial location dimension. The spatial dependence between positions is demonstrated by modeling the spatial attention diagram of any two positions in the feature graph, to highlight the wheel hub defect area in the global view and improve the detection effect of defects. • For cascade detectors, feature sharing is carried out in serial manner by decoupled classification and regression branches. • According to the actual production situation, the network structure is optimized, redundant parameters are reduced, and space is compressed. At present, object detection methods based on machine vision have been widely used in the field of industrial defect detection. Wheel hub defects are characterized by multiple scales and complex types. The location, size and affiliation of different defect marks are different, so it is difficult to establish an accurate wheel hub defect detection model. Therefore, a wheel nuclear hub defect detection method based on the DS-Cascade RCNN was proposed. To effectively locate multiscale s, a spatial attention mechanism was added as a wheel hub defect location enhancement module. Then deformable convolution is added, and the position and size of the convolution kernel are adjusted dynamically according to the shape of wheel defects. Finally, the pruning algorithm is used to optimize the improved model and compress the model space without losing the accuracy. The model was evaluated under the wheel dataset. Experimental results show that the proposed method can effectively detect six kinds of wheel hub defects, and the mean Average Precision (mAP) is 95.49%. Multiscale defect location and defect category estimation are realized, which meets the requirements of wheel hub detection in actual production. [ABSTRACT FROM AUTHOR]