PyTorch Practice Notebooks
-
Updated
Sep 12, 2021 - Jupyter Notebook
PyTorch Practice Notebooks
This project uses a TinyVGG16-based CNN to classify MRI scans for Alzheimer's Disease stages: Mild Impairment, Moderate Impairment, No Impairment, and Very Mild Impairment. It includes Jupyter notebooks for training and prediction, and a Streamlit app for easy inference. The model achieves high metrics in predicting Alzheimer's stages.
Add a description, image, and links to the pytorch-cnn-classification topic page so that developers can more easily learn about it.
To associate your repository with the pytorch-cnn-classification topic, visit your repo's landing page and select "manage topics."