Brain Tumor MRI images have been classified into 4 classes by using Transfer Learning Models. The classes are-
- Glioma Tumor
- Meningioma tumor
- Pituatary tumor
- No tumor
Currently, the model is trained without data augmentation.
In the future, data augmentation will be used on the training images to enhance the metrics.
Dataset Link - Dataset Link
Sr. No. | ML Algorithm | Training Accuracy | Testing Accuracy |
---|---|---|---|
1 | Inception Resnet | 98.2% | 94.3% |
2 | Resnet 152V2 | 99.7% | 99.1% |