Skip to content

Metal Surface Defect Inspection through Deep Convolution Neural Network

Notifications You must be signed in to change notification settings

FantacherJOY/Metal-Surface-Defect-Inspection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 
 
 

Repository files navigation

Metal-Surface-Defect-Inspection

It's a deep learning project used to classify metal surface defects. Images were collected for this project from the NEU database containing six defect types. Below is the result, where the first name is the prediction from the model, and the second name is in the bracket true category of respective images. You can download the data set from here: https://www.kaggle.com/fantacher/neu-metal-surface-defects-data.

This dataset contains a single folder with 1800 data. If you want to implement this project the way I did, you first need to separate these datasets into six folders with 300 images of each. Then you are done; just run the Notebook into your data directory, and you will get the same results. Please also improve the model by integrating it with any transfer learning model.

If you feel like citing this work in a research article, you can cite from Google Scholar.

Test Results

Releases

No releases published

Packages

No packages published