search

IIITKalyaniFOSC / MediCare-Prime

Prediction or detection of various medical ailments
https://medicare-prime.herokuapp.com/
GNU General Public License v3.0
21 stars 24 forks source link

Adding Brain Tumor Prediction and Chest MRI Disease Prediction models. #89

Closed kanakmi closed 2 years ago

kanakmi commented 2 years ago

Hey maintainers

I am the one who opened a new PR for adding a Brain Tumor Prediction model a few months back and my PR never got merged and now I understand exactly why that happened. The model size was ~300 MB which is definitely not something that can be deployed on Heroku.

I have a much better proposal this time. I want to add a Brain Tumor Prediction model which accepts Brain MRI images and classifies them as Brain Tumor Found or Not. Additionally, I also want to add a Chest MRI diagnosis model that classifies chest images as COVID-19, Viral Pneumonia, or Normal. Both the models are around 2 MB in size and wouldn't really increase the load of the server and would be a great addition to this project. If assigned, I would also add the corresponding functions to the "app.py" file and it could be used in similar ways you are currently using the Malaria Prediction model. All the current requirements would remain the same and the only significant increase in Heroku's slug size would be of both the models (~5 MB).

Do give it a thought and let me know if I could add them.

Thank You Kanak

kanakmi commented 2 years ago

And yes one more thing... Both the models are built from scratch and perform really well with an accuracy of around 98%.