/chest-X-rays-multiclass-densenet-attention

Pytorch code for multi-label classification of chest X-rays with DenseNet, ResNet and DenseNet-121 with attention mechanism. Heatmap generation using Grad-CAM has also been implemented for DenseNet-121 and DenseNet-121 with attention model

Primary LanguagePython

No issues in this repository yet.