Repository for Nuclei Segmentation for Histopathology Images

Note: If you're interested in using it, feel free to ⭐️ the repo so we know!

Datasets

CPM 15

The dataset can be downloaded from the link

CPM 17

The dataset can be downloaded from the link

Consep

A dataset that contains manually annotated 24,319 nuclei with associated class labels. The dataset can be downloaded from the link

Nuclei Segmentation

The dataset consist of 143 images ER+ BCa images scanned at 40x. Each image is 2,000 x 2,000. Across these images there are about 12,000 nuclei manually segmented. The dataset can be downloaded from the link

Patch Generation

Patch Generation has been done in offline mode to reduce pipeline complexity. The proposed approach uses multiple dataset therefore the size of WSIs are non-standard. Where as the pathches generated from WSIs have dimensions of 256x256x3 having 75% overlap among them.

Models

The purpose of this pipeline was to explore and train various nuclei segmentaion datasets therefore we used Modified U-Net for training.

U-Net

Pre-Trained Models

The Pre-Trained models can be downloaded from google drive.

Installation

To get this repo work please install all the dependencies using the command below:

pip install -U segmentation-models
pip install -r requirments.txt

Training

To start training run the Train.py script from the command below. For training configurations refer to the Training File file. You can update the file according to your training settings. Model avaible for training is U-NET.

 python Train.py

Testing

To perfrom Inference on the trained models on Test Images you first have to download the weights and place them in the results folder. After downliading the weights you unzip them and then run the Inference by using the command below.