Computer-Vision-Mathable/testare.py at master · tedi11/Computer-Vision-Mathable · 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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
def compare_annotations_position_token(filename_predicted, filename_gt, verbose=0):
    p = open(filename_predicted, "rt")
    gt = open(filename_gt, "rt")

    line_gt = gt.readline()
    current_pos_gt, current_token_gt = line_gt.split()
    if verbose:
        print(current_pos_gt, current_token_gt)

    line_p = p.readline()
    current_pos_p, current_token_p = line_p.split()
    if verbose:
        print(current_pos_p, current_token_p)

    match_positions = 1
    match_tokens = 1
    if (current_pos_p != current_pos_gt):
        match_positions = 0
    if (current_token_p != current_token_gt):
        match_tokens = 0

    points_positions = 0.025 * match_positions
    points_tokens = 0.01 * match_tokens

    return points_positions, points_tokens


# change this on your machine pointing to your results (txt files)
predictions_path_root = "output/"

# change this on your machine to point to the ground-truth test
gt_path_root = "antrenare/"


def compare_annotations_score(filename_predicted, filename_gt, verbose=0):
    p = open(filename_predicted, "rt")
    gt = open(filename_gt, "rt")

    all_lines_gt = gt.readlines()
    all_lines_p = p.readlines()

    number_lines_gt = len(all_lines_gt)

    points_scores = 0

    try:
        for i in range(0, number_lines_gt):
            line_gt = all_lines_gt[i]
            line_p = all_lines_p[i]
            if (line_gt == line_p):
                points_scores += 0.04
    except:
        pass

    return points_scores


# change this to 1 if you want to print results at each turn
verbose = 0
total_points = 0
for game in range(1, 5):
    # change this for game in range(1,5):
    points_score = 0
    for turn in range(1, 51):

        name_turn = str(turn)
        if (turn < 10):
            name_turn = '0' + str(turn)

        filename_predicted = predictions_path_root + str(game) + '_' + name_turn + '.txt'
        filename_gt = gt_path_root + str(game) + '_' + name_turn + '.txt'

        game_turn = str(game) + '_' + name_turn
        points_position = 0
        points_tokens = 0

        try:
            points_position, points_tokens = compare_annotations_position_token(filename_predicted, filename_gt,
                                                                                verbose)
        except:
            print("Pentru imaginea: ", game_turn, " functia de evaluare a intampinat o eroare")

        print("Imaginea: ", game_turn, "Puncte task 1: ", points_position, "Puncte task 2: ", points_tokens)

        total_points = total_points + points_position + points_tokens

    filename_predicted_scores = predictions_path_root + str(game) + '_scores' + '.txt'
    filename_gt_scores = gt_path_root + str(game) + '_scores' + '.txt'
    points_score = compare_annotations_score(filename_predicted_scores, filename_gt_scores, verbose=0)
    print("Puncte task 3 ", points_score)

    total_points = total_points + points_score
    print("Total puncte = ", total_points)

print(total_points)