An ameliorated deep dense convolutional neural network for accurate recognition of casting defects in X-ray images

摘要

Recognizing defects in X-ray images plays an important role in the detection of internal defects in titanium alloy castings. However, the existing manual defects recognition methods have common drawbacks such as unstable artificial recognition, misrecognition, huge workload, and low efficiency of recognition. To make up for the shortcomings, an ameliorated deep dense convolutional neural network (BX-Net) was presented to accurately recognize casting defects in X-ray images and effectively extract highly discriminative features of different categories. DenseNet121 was used as the backbone of BX-Net and the feature extracted by DenseNet121 was fully shared by the two inputs of a bilinear pooling layer. Transfer learning was applied to reduce the demand for data and hyperparameters’ tuning. The backbone of BX-Net was firstly trained on the ImageNet and then all layers of BX-Net was fine-tuned on nine-hundred X-ray images of TiAl aero casting components. Other six deep convolutional neural networks (DenseNet121, EfficientNetB4, EfficientNetB7, ResNet50, VGG16 and Xception) were also trained to be compared with the presented BX-Net. Two Support Vector Machines were trained on the LBP features data set and HOG features data set of nine-hundred X-ray images of TiAl aero casting components respectively Experiments comparing BX-Net with other six deep learning models and two machine learning models on one hundred X-ray images (test set) of TiAl aero casting components were carried out. The comparison results show that BX-Net has the least parameters except for the Densenet121. The recall and accuracy of BX-Net were 99% and 99% respectively. In addition, the comparison results also show that BX-Net was the only model that learned discriminative feature representation of the casting X-ray image data set. The BX-Net proposed in this paper is expected to overcome the shortcomings of manual defects recognition.