T2_DSBA_Graphical_Methods/main.py at feature_final_cli · ArindamRoy23/T2_DSBA_Graphical_Methods · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
"""

"""

from src.parser.SVCDArgParser import *
from src.experiments.Experiments import *

def main():
    """

    """
    SVCDParser = SVCDArgParser()
    parsed_arguments = SVCDParser()
    experiment = Experiment()


    target_image_path = parsed_arguments.target_image
    scribble_path = parsed_arguments.scribble
    lambda_ = parsed_arguments.lmbd
    gamma = parsed_arguments.gamma
    alpha = parsed_arguments.alpha
    sigma = parsed_arguments.sigma
    tau_primal = parsed_arguments.tau_primal
    tau_dual = parsed_arguments.tau_dual
    max_iter = parsed_arguments.max_iter
    early_stop = parsed_arguments.early_stop
    tolerance = parsed_arguments.tolerance
    use_tqdm = parsed_arguments.use_tqdm


    segmenter = experiment.run_segmentation(
        target_image_path,
        scribble_path,
        lambda_ = lambda_,
        gamma = gamma,
        alpha = alpha,
        sigma = sigma,
        tau_primal = tau_primal,
        tau_dual = tau_dual,
        max_iter = max_iter,
        early_stop = early_stop,
        tolerance = tolerance,
        use_tqdm = use_tqdm
    )

    final_segmentation = np.argmax(segmenter.theta_t1, axis = 0).transpose(1, 0)

    image_id = experiment.find_image_id(target_image_path)

    experiment.save_segmentation_mask(final_segmentation, image_id)

if __name__ == '__main__':
    # print('Start')
    main()