This project classifies smoking images using VGG19 with data augmentation, and Captum for model explainability, identifying key features per prediction.
python data-science machine-learning deep-neural-networks deep-learning grad-cam pytorch image-classification transfer-learning binary-classification data-augmentation occlusion integrated-gradients captum vgg19-model primary-attributions layer-attributions
-
Updated
Oct 31, 2024 - Jupyter Notebook