Skip to content

Conformist101/BrainMRI_Scan_FastAI_segmentation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 

Repository files navigation

# BrainMRI_Scan_FastAI_segmentation

Objective: To use FASTAI unet learner to perform an image segmentation task i.e. to identify tumours from MRI of Brain. Also to log the loss metrices in Neptune AI logger and compare the results after hyperparameter tuning

                

Model Architecture: 

We will be using the DynamicUnet architecture for the following from fastai.visionThis U-Net will sit on top of an encoder (that can be a pretrained model) and with a final output of n_classes. During the initialization, it uses Hooks to determine the intermediate features sizes by passing a dummy input through the model and create the upward path automatically.

Pretrained model: resnet34 

metrics: DICE

weight_decay(wd): 1e-1

What is Neptune AI ?

Experiment management tool bringing organization and collaboration to data science projects.link: https://neptune.ai/

 

Result:

 

# Publication : https://medium.com/analytics-vidhya/identifying-brain-tumor-from-mri-images-using-fastai-and-metrics-tracking-using-neptune-ai-71fbc56febba

# Dataset : https://www.kaggle.com/mateuszbuda/lgg-mri-segmentation (by :Mateusz Buda)

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published