Пример #1
0
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:
Пример #2
0
def configure_trainer(config: Configuration, args: argparse.Namespace) -> bool:
    if args.trainer:
        config.trainer = True

    return True