Pneumonia Detection CNN
Pneumonia Detection CNN
Deep learning model using ResNet-50 for automated pneumonia detection from chest X-ray images. Achieves high accuracy for assisting radiologists in diagnosis.
Description
This application uses convolutional neural networks to classify chest X-ray images for pneumonia detection. The model assists radiologists in predicting abnormalities with high accuracy.
Model Architecture
- Base Model: ResNet-50 (pretrained on ImageNet)
- Framework: MXNet with GluonCV
- Transfer Learning: Fine-tuned for medical imaging
Dataset
- Normal vs Pneumonia classification
- Training, validation, and test splits
Setup
pip install mxnet-cu101
pip install gluoncv
pip install comet-ml
Features
- Automated X-ray classification
- Confusion matrix visualization via Comet.ml
- GPU acceleration support
- Transfer learning from ImageNet