Imported from https://github.com/pykao/ABCD-MICCAI2019