3D Brain Tumor Segmentation in PyTorch using U-Net & Eigenvector projection for color invariance [PROJECT]

[https://github.com/Saswatm123/3D-Brain-Tumor-Segmentation-PyTorch](https://github.com/Saswatm123/3D-Brain-Tumor-Segmentation-PyTorch)

**Preface: I'm looking for a CV or ML/MLE job since getting my last offer rescinded in December.**

I created a 3D Brain Tumor segmentation model using PyTorch that takes in human brain MRI scan data as input, and outputs a segmentation map of the part of the brain that has a tumor, if it is present.

[Input images example](https://i.redd.it/ajpif1dnk7ab1.gif)

​

[Predicted segmentation map for previous input](https://i.redd.it/hgsk2hxrk7ab1.gif)

There are many more pictures/GIFs on the README, but I would like to …

brain data human job machinelearning map mle mri part pytorch segmentation