/KL-Severity-Grading-Focal-Loss-Optimization-Grad-CAM

KL severity grading using SE-ResNet and SE-DenseNet architectures trained with Cross Entropy loss and Focal Loss. The hyperparameters of focal loss have been fine-tuned as well. Further, Grad-CAM has been implemented for visualization purposes.

Primary LanguageJupyter Notebook

Watchers