GCN-Based-Interactive-Prostate-Segmentationon-on-MR-Images

PyTorch code for Graph convolutional network based interactive prostate segmentation on MR images

Note

The pretrained model file MS_DeepLab_resnet_pretained_VOC.pth can be download.

Segmentation Result

Train

• Run the run_train.sh shell script with,

sh run_train.sh

Test

• Run the run_prediction.sh shell script with,

sh run_prediction.sh