rae-t627 / 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.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

rae-t627/KL-Severity-Grading-Focal-Loss-Optimization-Grad-CAM Stargazers