main.py

import argparse
import os
import yaml
from preprocessing import preprocessing
from prediction import predict
from postprocessing import postprocess
from shape_measurement import measure_shape
from Utils.utils import dict_to_namespace


def main():

    # INFERENCE

    # Select config file from argparse
    parser = argparse.ArgumentParser()
    parser.add_argument('--config', type=str, required=False)
    args = parser.parse_args()

    if args.config is None:
        args.config = "config/config.yaml"
    
    # navigate to config file
    config_path = os.path.join(os.path.dirname(__file__), args.config)

    # Load parameters from yaml file
    with open(config_path, 'r') as f:
        config = yaml.safe_load(f)
    
    config = dict_to_namespace(config)

    # ------ Preprocessing ------
    preprocessing(config)

    # ------ Segmentation/Prediction of ascending aorta ------
    predict(config)

    # ------ Postprocessing ------
    postprocess(config)

    # ------ Shape Measurement ------
    measure_shape(config)



        





if __name__ == '__main__':
    main()