Metal casting defects frequently occur in the manufacturing industry. There exist numerous types of defects, each capable of adversely affecting the entire industry. Featured-based transfer learning has demonstrated potential in metal casting classification. In this study, we investigate the ability of different feature-based transfer learning pipelines in classifying casting defects. A total dataset of 400 images containing two class types, defective and normal was utilized in a ratio of 70:15:15 for training, validation and testing. The VGG16 pre-trained convolutional neural network model was used to extract the features from the images. Different classifiers, namely Support Vector Machine (SVM), k-Nearest Neighbours (kNN) and Logistic Regression (LR) were utilized to classify images based on the extracted features. The present study implies that VGG16 + LR was able to distinguish the classes well with a testing accuracy of 98%. The study paves the way to the possible implementation of the suggested pipelines in the classification of casting defects.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Contribution Title Optimizing Casting Defect Classification: A Comparative Analysis of Different Feature-Based Transfer Learning Pipelines

  • Chay Zheng Heng,
  • Anwar P.P. Abdul Majeed,
  • Saad Aslam,
  • Rabiu Muazu Musa,
  • Mehran Behjati,
  • Muhammed Basheer Jasser

摘要

Metal casting defects frequently occur in the manufacturing industry. There exist numerous types of defects, each capable of adversely affecting the entire industry. Featured-based transfer learning has demonstrated potential in metal casting classification. In this study, we investigate the ability of different feature-based transfer learning pipelines in classifying casting defects. A total dataset of 400 images containing two class types, defective and normal was utilized in a ratio of 70:15:15 for training, validation and testing. The VGG16 pre-trained convolutional neural network model was used to extract the features from the images. Different classifiers, namely Support Vector Machine (SVM), k-Nearest Neighbours (kNN) and Logistic Regression (LR) were utilized to classify images based on the extracted features. The present study implies that VGG16 + LR was able to distinguish the classes well with a testing accuracy of 98%. The study paves the way to the possible implementation of the suggested pipelines in the classification of casting defects.