scripts package#

Subpackages#

Submodules#

scripts.ae_bundle_streamlines module#

Script for used for the bundling of a whole-brain tractogram based on an autoencoder latent space

scripts.ae_bundle_streamlines.batch_filtering(args, batch_size, num_points, streamline_classes, device, model, latent_atlas_all, y_latent_atlas_all, thresholds)#
scripts.ae_bundle_streamlines.main()#
scripts.ae_bundle_streamlines.whole_filtering(args, num_points, streamline_classes, device, model, latent_atlas_all, y_latent_atlas_all, thresholds)#

scripts.ae_find_thresholds module#

Script used to automatically find the latent space optimal distance threshold.

scripts.ae_find_thresholds.main()#
scripts.ae_find_thresholds.separate_pl_impl(streamlines_dict: dict, key_list: List)#
scripts.ae_find_thresholds.separate_pl_impl_latent(latent_dict: dict, key_list: List, latent_dims: int = 32)#
scripts.ae_find_thresholds.set_threshold(config, streamlines_dict, latent_atlas_dict, latent_dict, streamline_classes, model, experiment_dir)#
scripts.ae_find_thresholds.sort_streamline_per_bundle(X: array, y: array) Tuple[array, array]#

scripts.ae_generate_streamlines module#

scripts.ae_train module#

Script for the inference and training of FINTA-multibundle

scripts.ae_train.main()#

scripts.fetch_data module#

scripts.fetch_data.main()#

Module contents#