smahesh2694 / NEU_XAI

Explainable AI algorithms on NEU steel surface defect dataset with Keras

NEU_XAI

This repo contains keras implementation of few XAI algorithms on NEU surface defect dataset.

Repo consists of :

1. NEU steel surface defect database --> Original 1800 image NEU dataset
2. NEU steel surface defect database - Test Split --> contains 180 test images (with 10% partition)
3. Finetune_on_NEU_dataset.ipynb --> i) Training a MobileNet model on the NEU surface defect dataset using Transfer Learning. ii) visualizing the important features of test images using model agnostic LIME and SHAP algorithms.

Sample images from NEU dataset :

Sample images of all 6 classes