/cAAE

code for Unsupervised Detection of Lesions in Brain MRI using constrained adversarial auto-encoders

Primary LanguagePythonMIT LicenseMIT

cAAE

code for Unsupervised Detection of Lesions in Brain MRI using constrained adversarial auto-encoders, https://arxiv.org/abs/1806.04972

AAE model is implemented based on this github repo

Required: python>=2.7, tensorlayer, tensorflow>=1.0, numpy

train model: python main.py --model_name "cAAE" --z_dim "128"