https://github.com/frozenwolf-cyber/brain-tumor-classification

Fine-tuning pre-trained ResNet50 model to classify the brain scan different types of tumor
https://github.com/frozenwolf-cyber/brain-tumor-classification

classification cnn deep-learning neural-network pytorch resnet-50 tensorflow2 torchvision

Last synced: 4 months ago
JSON representation

Fine-tuning pre-trained ResNet50 model to classify the brain scan different types of tumor

• Host: GitHub
• URL: https://github.com/frozenwolf-cyber/brain-tumor-classification
• Owner: FrozenWolf-Cyber
• Created: 2021-10-15T17:44:47.000Z (almost 4 years ago)
• Default Branch: main
• Last Pushed: 2021-12-06T05:17:52.000Z (almost 4 years ago)
• Last Synced: 2025-02-27T06:32:23.615Z (8 months ago)
• Topics: classification, cnn, deep-learning, neural-network, pytorch, resnet-50, tensorflow2, torchvision
• Language: Jupyter Notebook
• Homepage:
• Size: 2.16 MB
• Stars: 1
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

          
# Brain Tumor Classification

Fine-tuning pre-trained ResNet50 model to classify the brain scan among different types of tumors : Meningioma , Glioma and Pitutary.

To checkout Kaggle Notebook click [here](https://www.kaggle.com/frozenwolf/brain-tumor-classification)

## Sample Dataset:

![sample_dataset](https://user-images.githubusercontent.com/57902078/137530839-a608e20f-75bc-4b45-9750-afb7cf737404.png)

![sample_dataset1](https://user-images.githubusercontent.com/57902078/137530946-f86589b5-d873-4b0e-94f3-23df092f7244.png)

![sample_dataset2](https://user-images.githubusercontent.com/57902078/137530960-1c16a796-ea8a-4be6-8a0d-5c9315c43d35.png)

## Results :

### ResNet50 :

Test Accuracy :   96.875  

Test Loss :       0.1528246495872736

- #### Loss :

 ![loss](https://user-images.githubusercontent.com/57902078/137531104-36d1cbce-ae34-45dd-8c1f-4e6ac1e0afdf.png)

- #### Accuracy :

![accuracy](https://user-images.githubusercontent.com/57902078/137531108-c6f1ac33-da7b-430e-881b-2754b733501d.png)