/Neural_diffusion_distance

This is an implementation for training neural diffusion distance

Primary LanguagePython

Neural_diffusion_distance

This is an implementation for training neural diffusion distance. Please refer to

 Jian Sun, Zongben Xu. Neural diffusion distance for image segmentation, NeurIPS 2019.

Please download the following files.

(1) Initial model: InitModel.zip

https://drive.google.com/open?id=1DxqSBAoPylQ23bINWmXfSPezbeSGd5PL

Copy the 'InitModel' folder into the folder of 'DiffusionMap'

(2) Trained neural diffusion distance network: snapshots_diffMap_grid.zip

https://drive.google.com/open?id=1tvvgTl4t6DkHrf2whK12ewo7ATbKWMxi

Copy the 'snapshots_diffMap_grid' folder into the folder of 'DiffusionMap'

(3) BSD500 training dataset: Dataset.zip

https://drive.google.com/open?id=1bUwYb-RAnF1YTi1Bnmw2La2BujQLVFlE

Copy the 'Dataset' folder to the same directory of 'DiffusionMap', i.e., making sure that 'Dataset' and 'DiffusionMap' are in the same directory.


Please follow "README.txt" in folder of "DiffusionMap" for training neural diffusion network.