Skip to content

wentaozhu/adversarial-deep-structural-networks

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

adversarial-deep-structural-networks

Adversarial Deep Structural Networks for Mammographic Mass Segmentation https://arxiv.org/abs/1612.05970 The overview is in isbi18_poster.pptx

This is the first work formulating the adversarial example to improve segmentation. Three aspects to understand adversarial deep structure model 1) optimal regularization for local smoothness 2) optimal data augmentation 3) force the model learn well on the boundary area.

crfcnn_combine.py is the executive file.

modelname = 'cnn' #'crfcomb'

modelname = 'cnn' #is for FCN model.

modelname = 'cnnat' #is for advesarial FCN model.

modelname = 'crf' #is for joint FCN-CRF model.

modelname = 'crfat' #is for adversarial FCN-CRF model.

modelname = 'cnncomb' #is for multi-FCN model.

modelname = 'cnncombat' #is for adversarial multi-FCN model.

modelname = 'crfcomb' #is for joint multi-FCN-CRF model.

modelname = 'crfcombat' #is for adversarual multi-FCN-CRF model.

crfcnn_combine_ddsm.py is the main file for ddsm.

.m files are for reproducing miccai 15 Deep Learning and Structured Prediction for the Segmentation of Mass in Mammograms.

The inbreast dataset can be downloaded from https://drive.google.com/a/uci.edu/file/d/0B-7-8LLwONIZM1djY2pRLWNUemc/view?usp=sharing

The DDSM-BCRP dataset can be downloaded from https://drive.google.com/a/uci.edu/file/d/0B-7-8LLwONIZU0l2N3hXdU96Y2M/view?usp=sharing

Please cite our paper as Zhu, Wentao, and Xiaohui Xie. "Adversarial deep structural networks for mammographic mass segmentation." arXiv preprint arXiv:1612.05970 (2016).

If you have any questions, please contact with me wentaozhu1991@gmail.com.

Supplement code and data in https://drive.google.com/file/d/0B5Hl9mO74DHvUEowa1hyWmVsMmc/view?usp=sharing . Maybe it is helpful for you to reproduce the results.

About

ISBI2018: Adversarial Deep Structural Networks for Mammographic Mass Segmentation https://arxiv.org/abs/1612.05970

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published