wyf0912 / LDDG

[NIPS 2020] The code release of paper 'Domain Generalization for Medical Imaging Classification with Linear-Dependency Regularization'
61 stars 9 forks source link

Domain Generalization for Medical Imaging Classification with Linear Dependency Regularization

The code release of paper 'Domain Generalization for Medical Imaging Classification with Linear-Dependency Regularization' NeurIPS 2020. The pre-print paper can be found in Arxiv.

How to use

First, you need to install the package surface-distance https://github.com/deepmind/surface-distance and SimpleITK

pip install SimpleITK

Then run the following command to train and evaluate the performance of the model

python3 train_lddg.py -t i

where i means set_i is the target domain.

Segmentation Reuslts

image

Please cite our paper if you find it's useful.