Dataset has been prepared using roboflow on 100 mris of brain. It aims to recognize 2 parts, mid brain and pons.

To use the trained model simply download the "PD_predict.ipynb" and the trained model "best.pt"