SrijanShovit / HealthLearning

A repo comprising of various Machine Learning and Deep Learning projects in healthcare domain.

38 stars 53 forks source link

[Project Addition]: Brain Tumor Segmentation #135

Closed Karthik110505 closed 3 months ago

Karthik110505 commented 4 months ago

🔴 Project Title : Brain Tumor Segmentation

🔴 Aim : To Segment the tumor in brain

🔴 Dataset :https://www.kaggle.com/datasets/nikhilroxtomar/brain-tumor-segmentation?select=masks

🔴 Approach : Use the pretrained models like variants of yolo inorder to segment the brain tumor .

@SrijanShovit Please assign me this issue

github-actions[bot] commented 4 months ago

Congratulations, @Karthik110505! 🎉 Thank you for creating your issue. Your contribution is greatly appreciated and we look forward to working with you to resolve the issue. Keep up the great work!

We will promptly review your changes and offer feedback. Keep up the excellent work! Kindly remember to check our contributing guidelines

SrijanShovit commented 4 months ago

What all steps would you apply. Provide a detailed roadmap(including optim, act functions, metrics,etc)

Karthik110505 commented 4 months ago

Process that I follow:

Download the dataset from kaggle
Inspect the data and prepare it for model training
Try downloading yolo varients like yolov8 , yolov7, yolov5 and train the model.
Evaluate the model performance and modify the parameters for better model performance
Choose the best performing model
Predict the brain tumor in testing dataset
The output consist of images with segmentation around the brain tumor. Note: Currently I can't mention the optm, act and metrics . As we can't tell the perfect one's before the model training. They vary with data that we take

@SrijanShovit

github-actions[bot] commented 3 months ago

This issue has been automatically closed because it has been inactive for more than 7 days. If you believe this is still relevant, feel free to reopen it or create a new one. Thank you!