rae-t627 / kl-severity-grading-focal-loss-optimization-grad-cam Goto Github PK
View Code? Open in Web Editor NEWKL 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.