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()#