Index

A

All (Utils.training_utils.CheckpointingStrategy attribute)

apply_blacklists() (in module Utils.training_utils) apply_logging_config() (in module Utils.logging_cfg)

B

Best (Utils.training_utils.CheckpointingStrategy attribute)

BinaryClassificationEvaluator (class in Trainer.evaluation)

C

CachingMode (class in Pipeline.TorchDataGeneratorUtils.torch_internal) calc_ccc_global() (in module Utils.eval_utils) calc_ccc_mean() (in module Utils.eval_utils) CheckpointingStrategy (class in Utils.training_utils) commit() (Trainer.evaluation.BinaryClassificationEvaluator method) (Trainer.evaluation.Evaluator method) (Trainer.evaluation.SensorToFlowfrontEvaluator method)

ComplexListLoopingStrategy (class in Pipeline.TorchDataGeneratorUtils.looping_strategies) count_parameters() (in module Utils.training_utils) create_list_and_test() (Tests.test_data_loaders_IMG.TestDataLoaderIMG method) create_np_image() (in module Utils.img_utils) create_optimizer_and_lr_scheduler() (Trainer.ModelTrainer.ModelTrainer method) create_triangle_mesh() (in module Utils.dry_spot_detection_leoben)

D

DataloaderDryspots (class in Pipeline.data_loader_dryspot) DataloaderImages (class in Pipeline.data_loaders_IMG) DataloaderImageSequences (class in Pipeline.data_loaders_IMG) DataLoaderListLoopingStrategy (class in Pipeline.TorchDataGeneratorUtils.looping_strategies) DataLoaderSensor (class in Pipeline.data_loader_sensor) DeconvModel (class in Models.erfh5_DeconvModel) DeconvModel2x (class in Models.erfh5_DeconvModel) DeconvModel4x (class in Models.erfh5_DeconvModel)

DeconvModel8x (class in Models.erfh5_DeconvModel) DeconvModelEfficient (class in Models.erfh5_DeconvModel) DeconvModelEfficientBn (class in Models.erfh5_DeconvModel) discover() (Pipeline.TorchDataGeneratorUtils.torch_internal.FileDiscovery method) dry_spot_analysis() (in module Utils.dry_spot_detection_leoben) DrySpotModel (class in Models.erfh5_ConvModel) dump_content() (Pipeline.TorchDataGeneratorUtils.looping_strategies.DataLoaderListLoopingStrategy method) (Pipeline.TorchDataGeneratorUtils.looping_strategies.LoopingStrategy method)

E

erfh5_Autoencoder (class in Models.erfh5_autoencoder) erfh5_Conv25D_Frame (class in Models.erfh5_ConvModel) erfh5_Conv2dPercentage (class in Models.erfh5_ConvModel) erfh5_Conv3d (class in Models.erfh5_ConvModel) erfh5_Distributed_Autoencoder (class in Models.erfh5_autoencoder) ERFH5_Pressure_CNN (class in Models.erfh5_pressuresequence_CRNN)

ERFH5_Pressure_FC (class in Models.erfh5_pressuresequence_CRNN) ERFH5_PressureSequence_Model (class in Models.erfh5_pressuresequence_CRNN) ERFH5_RNN (class in Models.erfh5_pressuresequence_CRNN) (class in Models.erfh5_RNN_models) eval_preparation() (in module Utils.eval_utils) Evaluator (class in Trainer.evaluation)

F

FileDiscovery (class in Pipeline.TorchDataGeneratorUtils.torch_internal) FileList (Pipeline.TorchDataGeneratorUtils.torch_internal.CachingMode attribute) FileSetIterable (class in Pipeline.TorchDataGeneratorUtils.torch_internal) FileSetIterator (class in Pipeline.TorchDataGeneratorUtils.torch_internal) flip_array_diag() (in module Utils.img_utils) FlowfrontCNN (class in Models.flow_front_to_fiber_fraction_model) FlowfrontFeaturesRNN (class in Models.flow_front_to_fiber_fraction_model) FlowfrontToFiberfractionModel (class in Models.flow_front_to_fiber_fraction_model) focal_loss() (in module Models.custom_loss) FocalLoss (class in Models.custom_loss) forward() (Models.custom_loss.FocalLoss method) (Models.erfh5_ConvModel.DrySpotModel method) (Models.erfh5_ConvModel.S20Channel4toDrySpot method) (Models.erfh5_ConvModel.S20DeconvToDrySpotEff method) (Models.erfh5_ConvModel.S20DeconvToDrySpotEff2 method) (Models.erfh5_ConvModel.S20DeconvToDrySpotEff3 method) (Models.erfh5_ConvModel.S80Deconv2ToDrySpotEff method) (Models.erfh5_ConvModel.S80DeconvToDrySpotEff method) (Models.erfh5_ConvModel.SensorDeconvToDryspot method) (Models.erfh5_ConvModel.SensorDeconvToDryspot2 method) (Models.erfh5_ConvModel.SensorDeconvToDryspotEfficient method) (Models.erfh5_ConvModel.SensorDeconvToDryspotEfficient2 method) (Models.erfh5_ConvModel.SensorToDryspotBoolModel method) (Models.erfh5_ConvModel.erfh5_Conv25D_Frame method) (Models.erfh5_ConvModel.erfh5_Conv2dPercentage method) (Models.erfh5_ConvModel.erfh5_Conv3d method) (Models.erfh5_DeconvModel.DeconvModel method) (Models.erfh5_DeconvModel.DeconvModel2x method) (Models.erfh5_DeconvModel.DeconvModel4x method) (Models.erfh5_DeconvModel.DeconvModel8x method) (Models.erfh5_DeconvModel.DeconvModelEfficient method) (Models.erfh5_DeconvModel.DeconvModelEfficientBn method) (Models.erfh5_DeconvModel.S20DeconvModelEfficient method) (Models.erfh5_DeconvModel.S80DeconvModelEfficient method) (Models.erfh5_DeconvModel.S80DeconvModelEfficient2 method) (Models.erfh5_RNN_models.ERFH5_RNN method) (Models.erfh5_RNN_models.Net method) (Models.erfh5_autoencoder.StackedFullyConnected method) (Models.erfh5_autoencoder.erfh5_Autoencoder method) (Models.erfh5_autoencoder.erfh5_Distributed_Autoencoder method) (Models.erfh5_fullyConnected.S1140DryspotModelFCWide method) (Models.erfh5_fullyConnected.S20DSModelFC4ChannelsFlatten method) (Models.erfh5_fullyConnected.S20DryspotModelFCWide method) (Models.erfh5_fullyConnected.S80DryspotModelFCWide method) (Models.erfh5_fullyConnected.SensorDryspotModelFC method) (Models.erfh5_pressuresequence_CRNN.ERFH5_PressureSequence_Model method) (Models.erfh5_pressuresequence_CRNN.ERFH5_Pressure_CNN method) (Models.erfh5_pressuresequence_CRNN.ERFH5_Pressure_FC method) (Models.erfh5_pressuresequence_CRNN.ERFH5_RNN method) (Models.flow_front_to_fiber_fraction_model.FlowfrontCNN method) (Models.flow_front_to_fiber_fraction_model.FlowfrontFeaturesRNN method) (Models.flow_front_to_fiber_fraction_model.FlowfrontToFiberfractionModel method)

G

get_all_data_paths() (in module Resources.training) get_all_sequences_for_file() (in module Pipeline.data_loaders) get_data_paths() (in module Resources.training) get_data_paths_base_0() (in module Resources.training) get_data_paths_base_200k() (in module Resources.training) get_data_paths_debug() (in module Resources.training) get_data_paths_new_test_set() (in module Resources.training) get_encoding() (Models.erfh5_autoencoder.erfh5_Autoencoder method) get_example_erfh5() (in module Resources.training) get_filelist_within_folder() (in module Pipeline.data_gather) get_filelist_within_folder_blacklisted() (in module Pipeline.data_gather) get_flowfront_bool_dryspot() (Pipeline.data_loader_dryspot.DataloaderDryspots method) get_folders_within_folder() (in module Pipeline.data_gather)

get_images_of_flow_front_and_permeability_map() (Pipeline.data_loaders_IMG.DataloaderImageSequences method) get_index_sequence() (in module Pipeline.data_loaders) get_more_data_paths() (in module Resources.training) get_new_iterator() (Pipeline.TorchDataGeneratorUtils.looping_strategies.ComplexListLoopingStrategy method) (Pipeline.TorchDataGeneratorUtils.looping_strategies.DataLoaderListLoopingStrategy method) (Pipeline.TorchDataGeneratorUtils.looping_strategies.LoopingStrategy method) (Pipeline.TorchDataGeneratorUtils.looping_strategies.NoOpLoopingStrategy method) (Pipeline.TorchDataGeneratorUtils.looping_strategies.SimpleListLoopingStrategy method) get_remaining_files() (Pipeline.TorchDataGeneratorUtils.torch_internal.FileSetIterator method) get_samples() (Pipeline.TorchDataGeneratorUtils.torch_internal.SubSetGenerator method) get_sensor_bool_dryspot() (Pipeline.data_loader_dryspot.DataloaderDryspots method) get_sensordata_and_filling_percentage() (Pipeline.data_loader_sensor.DataLoaderSensor method) get_sensordata_and_flowfront() (Pipeline.data_loaders_IMG.DataloaderImages method) get_single_states_and_fillings() (in module Pipeline.data_loaders)

I

inference_on_test_set() (Trainer.ModelTrainer.ModelTrainer method) init_hidden() (Models.erfh5_pressuresequence_CRNN.ERFH5_RNN method) (Models.erfh5_RNN_models.ERFH5_RNN method)

init_weights() (Models.erfh5_pressuresequence_CRNN.ERFH5_Pressure_FC method) (Models.erfh5_RNN_models.ERFH5_RNN method) (Models.erfh5_pressuresequence_CRNN.ERFH5_RNN method) (Models.flow_front_to_fiber_fraction_model.FlowfrontFeaturesRNN method)

L

load_content() (Pipeline.TorchDataGeneratorUtils.looping_strategies.DataLoaderListLoopingStrategy method) (Pipeline.TorchDataGeneratorUtils.looping_strategies.LoopingStrategy method) load_model() (Models.erfh5_ConvModel.SensorDeconvToDryspot2 method)

load_model_layers_from_path() (in module Models.model_utils) load_stacked_fc() (in module Models.erfh5_autoencoder) LoopingStrategy (class in Pipeline.TorchDataGeneratorUtils.looping_strategies)

M

main() (in module Utils.dry_spot_detection_leoben) mark_useless_frames() (in module Utils.useless_frame_detection) mark_useless_frames_in_file() (in module Utils.useless_frame_detection) Models (module) Models.custom_loss (module) Models.erfh5_autoencoder (module) Models.erfh5_ConvModel (module)

Models.erfh5_DeconvModel (module) Models.erfh5_fullyConnected (module) Models.erfh5_pressuresequence_CRNN (module) Models.erfh5_RNN_models (module) Models.flow_front_to_fiber_fraction_model (module) Models.model_utils (module) ModelTrainer (class in Trainer.ModelTrainer) multiprocess_wrapper() (in module Utils.dry_spot_detection_leoben)

N

natural_sort_key() (in module Utils.natural_sorting) Net (class in Models.erfh5_RNN_models)

NoOpLoopingStrategy (class in Pipeline.TorchDataGeneratorUtils.looping_strategies) Nothing (Pipeline.TorchDataGeneratorUtils.torch_internal.CachingMode attribute)

P

Pipeline.data_gather (module) Pipeline.data_loader_dryspot (module) Pipeline.data_loader_sensor (module) Pipeline.data_loaders (module) Pipeline.data_loaders_IMG (module) Pipeline.TorchDataGeneratorUtils.looping_strategies (module)

Pipeline.TorchDataGeneratorUtils.torch_internal (module) pixel_wise_loss_multi_input_single_label() (in module Trainer.evaluation) plot_predictions_and_label() (in module Trainer.evaluation) prepare_subset() (Pipeline.TorchDataGeneratorUtils.torch_internal.SubSetGenerator method) print_metrics() (Trainer.evaluation.BinaryClassificationEvaluator method) (Trainer.evaluation.Evaluator method) (Trainer.evaluation.SensorToFlowfrontEvaluator method)

R

read_cmd_params() (in module Utils.training_utils) reset() (Trainer.evaluation.BinaryClassificationEvaluator method) (Trainer.evaluation.Evaluator method) (Trainer.evaluation.SensorToFlowfrontEvaluator method) resize_label_if_necessary() (Trainer.ModelTrainer.ModelTrainer method)

Resources.testing (module) Resources.training (module) run_ccc_calculations() (in module Utils.eval_utils) run_eval_w_binary_classificator() (in module Utils.eval_utils) running_in_docker() (in module Resources.testing)

S

S1140DryspotModelFCWide (class in Models.erfh5_fullyConnected) S20Channel4toDrySpot (class in Models.erfh5_ConvModel) S20DeconvModelEfficient (class in Models.erfh5_DeconvModel) S20DeconvToDrySpotEff (class in Models.erfh5_ConvModel) S20DeconvToDrySpotEff2 (class in Models.erfh5_ConvModel) S20DeconvToDrySpotEff3 (class in Models.erfh5_ConvModel) S20DryspotModelFCWide (class in Models.erfh5_fullyConnected) S20DSModelFC4ChannelsFlatten (class in Models.erfh5_fullyConnected) S80Deconv2ToDrySpotEff (class in Models.erfh5_ConvModel) S80DeconvModelEfficient (class in Models.erfh5_DeconvModel) S80DeconvModelEfficient2 (class in Models.erfh5_DeconvModel) S80DeconvToDrySpotEff (class in Models.erfh5_ConvModel) S80DryspotModelFCWide (class in Models.erfh5_fullyConnected) SampleWrapper (class in Tests.test_torch_datagenerator) save_encoder() (Models.erfh5_autoencoder.erfh5_Distributed_Autoencoder method) save_img() (in module Trainer.evaluation) SensorDeconvToDryspot (class in Models.erfh5_ConvModel) SensorDeconvToDryspot2 (class in Models.erfh5_ConvModel) SensorDeconvToDryspotEfficient (class in Models.erfh5_ConvModel) SensorDeconvToDryspotEfficient2 (class in Models.erfh5_ConvModel) SensorDryspotModelFC (class in Models.erfh5_fullyConnected) sensorgrid_simulationsuccess() (Pipeline.data_loader_sensor.DataLoaderSensor method) SensorToDryspotBoolModel (class in Models.erfh5_ConvModel)

SensorToFlowfrontEvaluator (class in Trainer.evaluation) setUp() (Tests.test_data_loaders_IMG.TestDataLoaderIMG method) (Tests.test_dry_spot_detection.TestDrySpotDetectionLeoben method) (Tests.test_eval_train_deconv.TestEval method) (Tests.test_hdf5_writer.TestPyWriter method) (Tests.test_ok_notok_training.TestOkNotOkTraining method) (Tests.test_torch_datagenerator.TestFileDiscovery method) (Tests.test_torch_datagenerator.TestFileSetIterable method) (Tests.test_torch_datagenerator.TestFileSetIterator method) (Tests.test_torch_datagenerator.TestLoopingDatagenerator method) (Tests.test_torch_datagenerator.TestLoopingStrategies method) (Tests.test_torch_datagenerator.TestSubsetGenerator method) (Tests.test_training_dryspot_from_ff.TestTrainingDryspotFF method) SimpleListLoopingStrategy (class in Pipeline.TorchDataGeneratorUtils.looping_strategies) split_aux_dicts() (in module Pipeline.TorchDataGeneratorUtils.looping_strategies) stack_aux_dicts() (in module Pipeline.TorchDataGeneratorUtils.looping_strategies) StackedFullyConnected (class in Models.erfh5_autoencoder) start_training() (Trainer.ModelTrainer.ModelTrainer method) store() (Pipeline.TorchDataGeneratorUtils.looping_strategies.ComplexListLoopingStrategy method) (Pipeline.TorchDataGeneratorUtils.looping_strategies.DataLoaderListLoopingStrategy method) (Pipeline.TorchDataGeneratorUtils.looping_strategies.LoopingStrategy method) (Pipeline.TorchDataGeneratorUtils.looping_strategies.SimpleListLoopingStrategy method) SubSetGenerator (class in Pipeline.TorchDataGeneratorUtils.torch_internal)

T

tearDown() (Tests.test_data_loaders_IMG.TestDataLoaderIMG method) (Tests.test_dry_spot_detection.TestDrySpotDetectionLeoben method) (Tests.test_eval_train_deconv.TestEval method) (Tests.test_hdf5_writer.TestPyWriter method) (Tests.test_ok_notok_training.TestOkNotOkTraining method) (Tests.test_training_dryspot_from_ff.TestTrainingDryspotFF method) test_caching() (Tests.test_torch_datagenerator.TestFileDiscovery method) test_can_load_abs_paths_with_root() (Tests.test_torch_datagenerator.TestSubsetGenerator method) test_cant_load_rel_paths_without_root() (Tests.test_torch_datagenerator.TestSubsetGenerator method) test_dataloader() (Tests.test_torch_datagenerator.TestFileSetIterable method) test_dataloader_mp() (Tests.test_torch_datagenerator.TestFileSetIterable method) test_dry_spots() (Tests.test_dry_spot_detection.TestDrySpotDetectionLeoben method) test_dump_load() (Tests.test_torch_datagenerator.TestLoopingStrategies method) test_eval() (Tests.test_eval_train_deconv.TestEval method) test_eval_preparation() (Tests.test_eval_train_deconv.TestEval method) test_gathering() (Tests.test_torch_datagenerator.TestFileDiscovery method) test_get_fixed_number_of_elements_and_their_indices_from_various_sized_list() (Tests.test_data_loaders_IMG.TestDataLoaderIMG method) test_get_remaining_files() (Tests.test_torch_datagenerator.TestFileSetIterator method) test_hdf5_writing() (Tests.test_hdf5_writer.TestPyWriter method) test_ignore_useless() (Tests.test_data_loaders_IMG.TestDataLoaderIMG method) test_loading() (Tests.test_torch_datagenerator.TestFileSetIterator method) test_noop_strategy() (Tests.test_torch_datagenerator.TestLoopingStrategies method) test_save_load_training() (Tests.test_eval_train_deconv.TestEval method) test_simple() (Tests.test_torch_datagenerator.TestFileSetIterable method) test_split_save_load() (Tests.test_torch_datagenerator.TestSubsetGenerator method) test_split_size() (Tests.test_torch_datagenerator.TestSubsetGenerator method) test_splits_add_up() (Tests.test_torch_datagenerator.TestLoopingDatagenerator method) test_strategies() (Tests.test_torch_datagenerator.TestLoopingStrategies method)

test_training() (Tests.test_eval_train_deconv.TestEval method) (Tests.test_training_dryspot_from_ff.TestTrainingDryspotFF method) test_training_apex() (Tests.test_eval_train_deconv.TestEval method) test_training_load_optimizer() (Tests.test_eval_train_deconv.TestEval method) test_training_ok_notok() (Tests.test_ok_notok_training.TestOkNotOkTraining method) TestDataLoaderIMG (class in Tests.test_data_loaders_IMG) TestDrySpotDetectionLeoben (class in Tests.test_dry_spot_detection) TestEval (class in Tests.test_eval_train_deconv) TestFileDiscovery (class in Tests.test_torch_datagenerator) TestFileSetIterable (class in Tests.test_torch_datagenerator) TestFileSetIterator (class in Tests.test_torch_datagenerator) TestLoopingDatagenerator (class in Tests.test_torch_datagenerator) TestLoopingStrategies (class in Tests.test_torch_datagenerator) TestOkNotOkTraining (class in Tests.test_ok_notok_training) TestPyWriter (class in Tests.test_hdf5_writer) Tests.test_data_loaders_IMG (module) Tests.test_dry_spot_detection (module) Tests.test_eval_train_deconv (module) Tests.test_hdf5_writer (module) Tests.test_ok_notok_training (module) Tests.test_torch_datagenerator (module) Tests.test_training_dryspot_from_ff (module) TestSubsetGenerator (class in Tests.test_torch_datagenerator) TestTrainingDryspotFF (class in Tests.test_training_dryspot_from_ff) to_list() (Tests.test_torch_datagenerator.SampleWrapper method) Trainer.evaluation (module) Trainer.ModelTrainer (module) transform_to_tensor_and_cache() (in module Utils.training_utils)

U

Utils.dry_spot_detection_leoben (module) Utils.eval_utils (module) Utils.img_utils (module)

Utils.logging_cfg (module) Utils.natural_sorting (module) Utils.training_utils (module) Utils.useless_frame_detection (module)