config.generator_labelling_atlases = [
m + '_' + str(nbr_ex) for m in gen_maps.values()
]
# Generator
config.generator = gen_start
if config.generator_house_expo:
gen_map = '_house_expo'
config.generator_labelling_atlases = [gen_map]
elif gen_start:
gen_map = gen_maps[chosen_map]
config.generator_gen_type = gen_map
# Trainer
config.trainer = train_start
config.trainer_model = training_algo # Either BasicLSTMModule or CAE or LSTMCAEModel
config.trainer_custom_config = None
config.trainer_pre_process_data_only = False
config.trainer_bypass_and_replace_pre_processed_cache = False
# Analyzer
config.analyzer = analyzer_start
MainRunner(config).run()
# To brute force generate map from image
# map_gen_from_img = False
# if map_gen_from_img == True:
def configure_trainer(config: Configuration, args: argparse.Namespace) -> bool:
if args.trainer:
config.trainer = True
return True