Imported from https://github.com/MRegina/connectome_conv_net