Skip to content

Latest commit

 

History

History
43 lines (35 loc) · 1008 Bytes

README.md

File metadata and controls

43 lines (35 loc) · 1008 Bytes

Brain_Tumour_Segmentation_3D_MRI

Implementation of Brain Tumor Segmentation of 3D MRI Imagery in Pytorch.

Colab Demo Run [Open In Colab](https://colab.research.google.com/drive/1KIk6CWy8G-1q2Z3KXdtE4-iOOOeZV-8-?usp=sharing]

Diretory Structre

├── src
│   ├── data.py
│   ├── logger.py
│   ├── loss.py
│   ├── metrics.py
│   ├── models.py
│   ├── train.py
│   ├── utils.py
│   └── val.py
├── LICENSE
├── README.md
├── main.py
└── requirements.txt

ToDo:

  • Research - Done
  • Setup Directory Structre and Github - Done
  • Solve for Data Download, Extract and Process - Done
  • Dataset Class - Done
  • Build 3D U net Model -Done
  • Define Loss Function - Done
  • Define Trainer - Done
  • Define Validation -Done
  • Define Logger
  • Train the model
  • Integrate wandb
  • Observe and Fine Tune
  • ReadMe
  • Things to improve