/Mask-RCNN-1

Nuclei identification using instance segmentation

Primary LanguagePythonOtherNOASSERTION

Mask-RCNN

This is an implementation of Mask RCNN for carrying out nuclei segmentation. The code has been referred from the Mask RCNN implementation by matterport.

Dataset

The dataset for Kaggle Data Science Bowl 2018 has been used here. The dataset must be put into the folder samples/nucleus/data/.

Training

Train a new model starting from ImageNet weights

cd samples/nucleus/
python3 nucleus.py train --dataset=data/ --subset=train --weights=imagenet

Train a new model starting from specific weights file

cd samples/nucleus/
python3 nucleus.py train --dataset=data/ --subset=train --weights=/path/to/weights.h5

Resume training a model that you had trained earlier

cd samples/nucleus/
python3 nucleus.py train --dataset=data/ --subset=train --weights=last

Generate submission file

cd samples
python3 nucleus_test.py 

Results

1 nucleus3_mrcnn 4