honglianghe / CDNet

ICCV2021-CDNet: Centripetal Direction Network for Nuclear Instance Segmentation

CDNet: Centripetal Direction Network for Nuclear Instance Segmentation

[ICCV2021]

The code includes training and inference procedures for CDNet.

Tips: There is a result written mistake (U-Net) in Table 4 in the original paper. The correct result is：

MoNuSeg

Method Name	Dice	AJI
U-Net	0.8184	0.5910
Mask-RCNN	0.7600	0.5460
DCAN	0.7920	0.5250
Micro-Net	0.7970	0.5600
DIST	0.7890	0.5590
CIA-Net	0.8180	0.6200
U-Net	0.8027	0.6039
Hover-Net	0.8260	0.6180
BRP-Net	-	0.6422
PFF-Net	0.8091	0.6107
Our CDNet	0.8316	0.6331

Getting Started

Create a data folder(/data) and put the datasets(MoNuSeg, CPM17) in it.

Train

cd CDNet/
python train.py

Test

cd CDNet/
python test.py