예제 #1
0
# 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:
#     Services = Services(config)
#     generated_map = Generator(Services)
#     generated_map.generate_map_from_image("map1.png",True,2)
예제 #2
0
def configure_analyzer(config: Configuration, args: argparse.Namespace) -> bool:
    if args.analyzer:
        config.analyzer = True

    return True