This project is about using Pytorch for 2D MRI image classification using transfer learning with Resnet34 with a testing accuracy is 98%.

The images are downloaded from Kaggle. These images will be classified into four categories, including

NOD: Non Demented,
VMD: Very Mild Demented,
MID: Mild Demented,
MOD: Moderate Demented. \

Index Description Jupiter notebook Content data
1 Val_accuracy: 0.98
Testing accuracy: 0.68 😢
01_Resrnet34.ipynb Download data
split to train, val, test
Train with Resnet34
Testing evaluation
Reasoning
org_day
2 Val_accuracy: 0.99
Testing accuracy: 0.98 😃
Is this approach ok? 🤔
02_Resnet34.ipynb Combine data
Split data
Train model
Evaluate the model
allnew

For utility functions, please see mymodulo.py