BlindNet/res.py at master · thunderock/BlindNet · 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
96
97
98
99
100
101
102
103
104
import matplotlib.pyplot as plt
import numpy as np
# all_scores = [{'total': 4952, 'pred_idx_mask': [55, 48, 30, 21, 18, 26, 99, 28, 48, 166, 18, 19, 56, 14, 19, 34, 45, 31, 36, 18, 45, 32, 32, 28, 39, 13, 41, 22, 40, 58, 30, 41, 43, 28, 40, 21, 50, 219, 47, 66, 538, 27, 52, 28, 33, 62, 67, 43, 28, 41, 24, 30, 49, 64, 27, 23, 25, 32, 19, 28, 38, 74, 73, 44, 50, 48, 64, 47, 96, 34, 30, 58, 44, 33, 21, 74, 91, 138, 251, 47, 38, 99, 56, 35, 71, 82, 29, 53, 43, 46, 41], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
# {'total': 4952, 'pred_idx_mask': [34, 44, 25, 22, 50, 30, 53, 46, 34, 52, 18, 25, 24, 32, 27, 39, 58, 31, 42, 18, 49, 60, 28, 23, 27, 20, 56, 27, 35, 67, 33, 63, 101, 49, 53, 30, 22, 329, 49, 38, 364, 30, 51, 27, 67, 60, 93, 27, 41, 56, 28, 27, 35, 70, 34, 30, 27, 33, 44, 24, 42, 31, 71, 36, 45, 17, 66, 57, 91, 37, 40, 61, 49, 30, 27, 56, 68, 118, 275, 29, 51, 84, 67, 29, 62, 66, 28, 52, 39, 174, 23], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
# {'total': 4952, 'pred_idx_mask': [41, 44, 20, 25, 34, 21, 43, 21, 30, 42, 44, 56, 31, 25, 19, 30, 38, 47, 30, 39, 51, 44, 35, 41, 34, 48, 68, 20, 56, 49, 33, 56, 61, 44, 39, 41, 15, 300, 49, 44, 126, 38, 37, 37, 45, 62, 479, 36, 27, 45, 27, 46, 33, 58, 34, 27, 38, 28, 41, 25, 53, 26, 71, 36, 52, 16, 65, 49, 100, 19, 23, 39, 40, 40, 32, 56, 61, 128, 327, 21, 37, 71, 66, 29, 30, 60, 32, 43, 31, 146, 56], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
# {'total': 4952, 'pred_idx_mask': [48, 49, 36, 40, 23, 23, 74, 46, 18, 68, 17, 53, 16, 36, 17, 43, 27, 41, 21, 32, 43, 50, 25, 27, 33, 21, 40, 19, 55, 80, 72, 70, 66, 49, 59, 27, 17, 267, 47, 49, 253, 34, 44, 26, 25, 58, 429, 48, 34, 34, 32, 23, 41, 64, 50, 15, 50, 59, 30, 30, 38, 36, 57, 33, 45, 34, 52, 27, 74, 18, 35, 37, 73, 48, 46, 41, 43, 123, 337, 7, 47, 34, 25, 74, 24, 50, 29, 30, 40, 117, 55], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
# {'total': 4952, 'pred_idx_mask': [51, 47, 34, 24, 11, 54, 85, 25, 22, 32, 18, 29, 11, 24, 50, 47, 29, 58, 31, 50, 47, 35, 20, 56, 30, 19, 62, 11, 15, 55, 48, 57, 55, 62, 103, 51, 26, 184, 64, 91, 382, 31, 47, 24, 38, 59, 433, 27, 22, 42, 44, 30, 27, 86, 46, 30, 41, 35, 35, 29, 32, 35, 67, 30, 40, 47, 46, 29, 60, 21, 27, 42, 48, 31, 49, 48, 74, 133, 302, 9, 43, 56, 23, 54, 29, 34, 30, 39, 32, 58, 53], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
# {'total': 4952, 'pred_idx_mask': [48, 46, 41, 21, 17, 65, 56, 39, 11, 43, 6, 26, 62, 35, 47, 39, 25, 67, 64, 29, 40, 39, 16, 61, 15, 14, 78, 21, 38, 80, 37, 33, 83, 88, 85, 57, 26, 133, 50, 70, 387, 28, 22, 28, 39, 67, 481, 26, 27, 33, 18, 32, 24, 72, 27, 53, 30, 59, 42, 40, 51, 40, 68, 35, 39, 35, 72, 17, 91, 20, 24, 47, 75, 19, 29, 41, 30, 136, 275, 16, 43, 36, 25, 34, 32, 34, 32, 33, 38, 53, 46], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
# {'total': 4952, 'pred_idx_mask': [63, 47, 30, 18, 26, 29, 54, 41, 22, 37, 15, 31, 9, 35, 38, 57, 30, 64, 45, 35, 42, 58, 22, 24, 20, 21, 68, 37, 24, 71, 53, 41, 91, 30, 72, 36, 18, 97, 36, 70, 619, 25, 52, 31, 28, 95, 139, 28, 24, 26, 35, 22, 25, 80, 17, 32, 38, 28, 39, 34, 34, 53, 65, 35, 51, 35, 126, 35, 105, 38, 38, 100, 67, 67, 23, 74, 44, 125, 271, 3, 50, 40, 31, 53, 22, 42, 37, 45, 28, 85, 56], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
# {'total': 4952, 'pred_idx_mask': [73, 40, 41, 24, 23, 20, 48, 44, 24, 50, 19, 25, 32, 31, 47, 72, 49, 57, 44, 30, 46, 63, 20, 62, 13, 13, 38, 15, 23, 71, 47, 36, 94, 43, 69, 56, 15, 94, 38, 107, 164, 23, 17, 26, 31, 54, 81, 29, 42, 27, 17, 10, 45, 55, 29, 32, 27, 45, 37, 50, 41, 40, 63, 39, 53, 30, 141, 27, 191, 66, 26, 73, 49, 45, 25, 42, 53, 118, 302, 9, 90, 31, 20, 53, 21, 55, 41, 35, 33, 511, 32], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
# {'total': 4952, 'pred_idx_mask': [73, 48, 32, 13, 25, 52, 47, 32, 31, 36, 22, 12, 45, 44, 43, 65, 29, 68, 36, 18, 54, 59, 20, 35, 29, 14, 47, 13, 26, 72, 51, 36, 83, 26, 101, 19, 17, 161, 82, 70, 698, 30, 23, 29, 27, 55, 31, 48, 39, 27, 17, 30, 53, 41, 68, 36, 25, 43, 30, 39, 36, 22, 69, 31, 54, 34, 177, 23, 93, 27, 14, 52, 48, 36, 61, 67, 41, 115, 341, 14, 42, 27, 28, 65, 22, 50, 36, 38, 33, 44, 37], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
# {'total': 4952, 'pred_idx_mask': [47, 42, 43, 28, 60, 42, 52, 46, 22, 33, 60, 22, 39, 26, 56, 29, 38, 73, 41, 21, 34, 41, 18, 21, 25, 14, 46, 12, 31, 79, 56, 31, 92, 28, 98, 17, 19, 105, 41, 73, 675, 27, 19, 28, 28, 54, 49, 31, 29, 35, 22, 27, 31, 49, 17, 32, 21, 75, 30, 33, 48, 39, 70, 44, 53, 26, 256, 24, 98, 28, 38, 89, 61, 31, 54, 40, 36, 109, 294, 36, 35, 31, 29, 75, 33, 41, 35, 46, 33, 68, 39], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
# {'total': 4952, 'pred_idx_mask': [72, 48, 42, 23, 26, 11, 59, 42, 21, 33, 56, 17, 21, 24, 74, 35, 25, 53, 42, 25, 48, 44, 21, 58, 22, 17, 56, 17, 18, 71, 47, 26, 94, 31, 67, 26, 23, 117, 52, 83, 667, 24, 19, 23, 20, 61, 29, 22, 38, 50, 27, 23, 49, 52, 17, 44, 18, 72, 34, 42, 31, 61, 70, 37, 45, 35, 135, 27, 150, 29, 65, 55, 52, 33, 44, 42, 28, 121, 306, 20, 60, 37, 46, 108, 27, 46, 43, 34, 33, 69, 45], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
# {'total': 4952, 'pred_idx_mask': [49, 43, 43, 26, 13, 54, 51, 51, 28, 43, 50, 20, 16, 23, 73, 67, 45, 32, 53, 22, 50, 53, 27, 16, 13, 23, 56, 15, 16, 73, 54, 52, 55, 39, 101, 52, 16, 138, 17, 59, 568, 27, 23, 22, 30, 90, 51, 28, 44, 32, 32, 27, 43, 43, 21, 36, 27, 57, 29, 31, 29, 57, 65, 55, 41, 55, 241, 24, 71, 55, 23, 50, 68, 22, 45, 96, 36, 115, 331, 23, 34, 41, 26, 95, 25, 44, 34, 30, 29, 75, 29], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
# {'total': 4952, 'pred_idx_mask': [72, 38, 34, 31, 24, 13, 58, 29, 60, 52, 64, 16, 44, 21, 71, 25, 33, 61, 55, 20, 48, 33, 25, 57, 20, 13, 43, 21, 34, 282, 34, 43, 49, 46, 82, 28, 25, 126, 43, 53, 146, 27, 8, 22, 26, 67, 86, 29, 33, 31, 29, 21, 44, 43, 38, 29, 26, 74, 49, 29, 44, 73, 78, 48, 37, 41, 467, 25, 104, 29, 18, 68, 56, 19, 49, 62, 30, 120, 289, 10, 39, 40, 31, 119, 26, 41, 37, 36, 26, 62, 45], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
# {'total': 4952, 'pred_idx_mask': [58, 42, 21, 24, 16, 20, 48, 30, 22, 35, 39, 19, 25, 21, 49, 68, 31, 71, 29, 22, 43, 50, 23, 23, 27, 22, 55, 19, 23, 92, 57, 42, 52, 38, 78, 31, 23, 180, 35, 73, 574, 27, 25, 19, 23, 48, 80, 36, 26, 48, 28, 20, 26, 53, 57, 43, 31, 86, 35, 26, 34, 63, 66, 37, 49, 54, 153, 23, 124, 28, 26, 48, 40, 37, 52, 61, 43, 106, 279, 28, 84, 35, 33, 149, 40, 48, 42, 40, 34, 63, 46], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]}]

all_scores = ([{'total': 4952, 'pred_idx_mask': [54, 60, 36, 34, 24, 18, 67, 14, 26, 24, 41, 25, 43, 52, 9, 40, 51, 70, 38, 64, 77, 73, 20, 50, 19, 13, 65, 33, 23, 45, 52, 38, 55, 27, 49, 27, 32, 52, 20, 87, 52, 65, 25, 60, 45, 74, 119, 32, 36, 27, 24, 35, 38, 46, 29, 33, 13, 46, 39, 34, 78, 40, 60, 35, 46, 35, 43, 66, 49, 43, 24, 53, 753, 34, 46, 88, 28, 115, 266, 29, 51, 30, 33, 20, 45, 60, 45, 84, 74, 28, 32], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},

{'total': 4952, 'pred_idx_mask': [58, 44, 30, 36, 45, 19, 68, 14, 24, 40, 27, 31, 45, 53, 30, 45, 40, 67, 29, 87, 85, 41, 29, 42, 17, 32, 37, 28, 36, 61, 75, 36, 53, 59, 52, 30, 50,
39, 24, 132, 38, 64, 27, 56, 48, 46, 116, 30, 28, 40, 22, 23, 28, 42, 33, 28, 21, 38, 29, 36, 83, 31, 75, 24, 50, 25, 36, 60, 95, 18, 29, 50, 674, 44, 39, 68, 33, 117, 274, 36, 54, 30, 21, 34, 68, 72, 38,
 99, 30, 38, 24], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},

{'total': 4952, 'pred_idx_mask': [72, 38, 32, 50, 21, 18, 52, 53, 44, 31, 21, 69, 63, 48, 41, 41, 53, 47, 48, 58, 34, 38, 22, 30, 25, 20, 32, 29, 24, 54, 77, 21, 72, 26, 71, 33, 57,
43, 25, 119, 29, 61, 22, 58, 26, 65, 160, 32, 30, 43, 30, 26, 28, 44, 31, 35, 16, 60, 34, 42, 51, 50, 59, 33, 64, 24, 44, 33, 73, 22, 42, 43, 109, 57, 611, 45, 34, 130, 276, 32, 43, 35, 30, 55, 29, 36, 40
, 116, 35, 51, 26], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},

{'total': 4952, 'pred_idx_mask': [54, 41, 42, 33, 19, 27, 50, 44, 37, 52, 35, 58, 76, 56, 33, 60, 22, 51, 48, 22, 50, 40, 39, 26, 22, 15, 49, 30, 16, 63, 32, 25, 80, 30, 85, 19, 70, 51, 18, 107, 37, 60, 31, 53, 51, 69, 136, 29, 44, 48, 22, 31, 40, 37, 39, 43, 20, 43, 36, 35, 114, 41, 79, 20, 48, 48, 46, 32, 72, 21, 45, 54, 53, 27, 535, 49, 26, 150, 257, 30, 82, 38, 16, 57, 27, 42, 31, 119, 34, 61, 67], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},

{'total': 4952, 'pred_idx_mask': [44, 45, 47, 33, 8, 32, 41, 53, 18, 43, 30, 53, 247, 67, 58, 35, 24, 45, 40, 34, 49, 34, 21, 21, 16, 16, 43, 20, 15, 54, 35, 28, 64, 40, 86, 23, 67, 62, 29, 82, 32, 56, 29, 95, 25, 60, 88, 21, 28, 29, 29, 12, 39, 37, 46, 57, 59, 43, 30, 36, 76, 34, 79, 36, 53, 43, 34, 26, 93, 30, 17, 42, 60, 43, 508, 43, 64, 114, 275, 48, 116, 39, 30, 42, 27, 59, 34, 90, 30, 51, 63], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},

{'total': 4952, 'pred_idx_mask': [61, 39, 37, 45, 20, 52, 49, 44, 22, 31, 70, 72, 192, 35, 41, 31, 24, 70, 42, 18, 28, 55, 45, 35, 23, 10, 47, 29, 13, 61, 47, 23, 66, 31, 100, 33, 52, 59, 27, 102, 40, 70, 40, 69, 35, 42, 140, 30, 41, 29, 25, 27, 41, 30, 54, 29, 37, 41, 31, 16, 402, 45, 56, 31, 47, 42, 39, 26, 219, 20, 14, 46, 42, 30, 104, 45, 61, 116, 266, 22, 109, 37, 63, 40, 17, 44, 27, 81, 27, 45, 41], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
{'total': 4952, 'pred_idx_mask': [59, 36, 33, 37, 22, 36, 48, 49, 15, 36, 54, 23, 197, 31, 64, 30, 31, 67, 42, 13, 39, 42, 26, 41, 20, 25, 50, 29, 53, 423, 49, 30, 59, 33, 54, 35, 43
, 72, 32, 95, 44, 35, 33, 60, 20, 60, 52, 25, 30, 29, 27, 42, 45, 34, 38, 31, 34, 73, 30, 34, 58, 39, 79, 17, 42, 26, 38, 27, 281, 73, 13, 75, 39, 28, 98, 70, 53, 127, 239, 36, 109, 43, 48, 39, 31, 52, 29
, 80, 29, 41, 44], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0
, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},

{'total': 4952, 'pred_idx_mask': [64, 36, 40, 27, 14, 29, 46, 38, 15, 31, 37, 17, 241, 38, 56, 79, 43, 67, 33, 14, 34, 44, 17, 66, 13, 11, 72, 21, 15, 142, 63, 66, 45, 37, 51, 27, 33
, 53, 19, 91, 52, 26, 20, 28, 40, 58, 62, 32, 33, 35, 26, 43, 21, 25, 49, 37, 25, 37, 35, 35, 32, 55, 68, 29, 37, 27, 38, 32, 537, 63, 18, 68, 44, 40, 94, 99, 39, 128, 283, 27, 87, 42, 61, 59, 38, 52, 34,
 82, 38, 42, 55], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},

{'total': 4952, 'pred_idx_mask': [93, 43, 35, 33, 33, 14, 51, 44, 9, 31, 57, 26, 181, 47, 31, 41, 28, 58, 49, 50, 43, 65, 21, 38, 25, 22, 48, 45, 14, 415, 42, 30, 59, 29, 68, 33, 50,
 47, 25, 69, 52, 23, 48, 32, 35, 90, 40, 26, 37, 38, 33, 43, 46, 55, 39, 28, 24, 30, 28, 20, 45, 58, 70, 18, 56, 42, 69, 17, 273, 55, 17, 53, 46, 23, 131, 96, 42, 107, 259, 19, 86, 39, 44, 78, 38, 65, 22,
 47, 33, 42, 53], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},

{'total': 4952, 'pred_idx_mask': [86, 44, 49, 29, 16, 19, 49, 41, 29, 34, 62, 20, 239, 71, 43, 78, 31, 72, 43, 8, 34, 46, 24, 66, 21, 26, 57, 56, 35, 415, 65, 31, 50, 33, 64, 22, 40,
 49, 26, 84, 77, 25, 26, 30, 15, 76, 34, 24, 35, 32, 33, 19, 40, 46, 40, 33, 23, 29, 49, 36, 64, 24, 68, 35, 43, 21, 66, 23, 234, 63, 15, 53, 47, 27, 82, 71, 28, 142, 258, 43, 78, 40, 38, 96, 24, 40, 32,
37, 40, 53, 38], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},

{'total': 4952, 'pred_idx_mask': [97, 53, 28, 21, 13, 15, 54, 42, 16, 42, 21, 21, 201, 16, 47, 73, 31, 35, 47, 8, 41, 48, 21, 57, 15, 20, 40, 53, 9, 86, 52, 26, 62, 36, 96, 24, 42, 77, 30, 97, 73, 22, 24, 28, 34, 68, 28, 30, 36, 35, 29, 30, 33, 35, 43, 26, 42, 32, 33, 19, 59, 46, 67, 53, 48, 57, 76, 25, 644, 49, 22, 52, 48, 41, 79, 65, 40, 121, 302, 32, 82, 54, 44, 67, 31, 41, 23, 50
, 25, 52, 44], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
{'total': 4952, 'pred_idx_mask': [86, 39, 33, 27, 20, 49, 62, 39, 56, 35, 30, 18, 260, 25, 33, 31, 57, 59, 47, 10, 30, 56, 31, 27, 14, 23, 47, 33, 34, 384, 54, 24, 47, 47, 57, 28, 29, 81, 22, 66, 57, 23, 30, 25, 28, 108, 44, 36, 38, 42, 47, 39, 39, 40, 35, 38, 21, 29, 45, 23, 61, 34, 75, 41, 43, 33, 55, 24, 247, 56, 25, 52, 32, 33, 91, 72, 33, 124, 276, 52, 82, 36, 35, 106, 30, 40, 44, 47, 30, 60, 46], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
{'total': 4952, 'pred_idx_mask': [308, 44, 24, 37, 20, 13, 50, 29, 28, 27, 23, 20, 85, 23, 37, 47, 32, 57, 51, 12, 32, 25, 15, 26, 12, 9, 194, 8, 18, 185, 57, 25, 53, 49, 54, 22, 35, 52, 21, 45, 38, 31, 13, 24, 37, 217, 54, 26, 42, 53, 42, 32, 37, 41, 50, 51, 33, 53, 65, 29, 32, 31, 63, 36, 61, 31, 177, 32, 63, 68, 44, 148, 58, 36, 48, 76, 49, 141, 297, 39, 39, 79, 53, 67, 35, 45, 33, 71, 33, 13, 82], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
{'total': 4952, 'pred_idx_mask': [111, 35, 16, 42, 32, 31, 46, 23, 32, 32, 22, 17, 79, 22, 28, 46, 33, 54, 35, 20, 29, 26, 22, 27, 13, 22, 104, 9, 29, 193, 235, 34, 56, 47, 56, 25, 12, 65, 25, 52, 41, 22, 20, 9, 37, 104, 46, 36, 49, 38, 35, 31, 28, 66, 54, 33, 48, 72, 62, 44, 51, 29, 56, 36, 56, 29, 98, 49, 219, 170, 38, 171, 77, 47, 27, 73, 64, 134, 275, 29, 35, 64, 56, 64, 53, 44, 40, 51, 35, 18, 52], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
{'total': 4952, 'pred_idx_mask': [297, 46, 38, 19, 28, 16, 64, 31, 22, 32, 39, 20, 73, 23, 56, 60, 27, 52, 33, 18, 38, 42, 22, 41, 28, 32, 60, 18, 42, 392, 60, 31, 44, 53, 111, 28, 47, 68, 21, 80, 34, 22, 16, 23, 34, 223, 32, 24, 30, 48, 50, 26, 33, 78, 35, 71, 40, 56, 47, 30, 36, 60, 79, 40, 45, 47, 55, 30, 67, 159, 31, 43, 53, 23, 29, 39, 34, 120, 280, 26, 62, 24, 35, 33, 28, 35, 48, 51, 34, 60, 42], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
{'total': 4952, 'pred_idx_mask': [106, 49, 31, 40, 24, 23, 54, 20, 37, 38, 29, 21, 93, 28, 34, 48, 30, 51, 34, 16, 24, 28, 23, 25, 12, 14, 164, 13, 32, 198, 192, 21, 46, 49, 34, 20, 15, 51, 17, 51, 48, 29, 19, 31, 33, 97, 56, 31, 57, 45, 50, 36, 40, 77, 50, 32, 26, 38, 53, 31, 35, 31, 71, 36, 69, 30, 61, 43, 217, 99, 59, 238, 74, 37, 46, 83, 53, 133, 295, 35, 42, 47, 48, 72, 30, 57, 35, 63, 31, 16, 52], 'pred_idx': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]},
{'total': 4952, 'pred_idx_mask': [74, 28, 32, 23, 28, 23, 24, 25, 24, 19, 32, 37, 66, 46, 34, 28, 38, 48, 43, 22, 21, 20, 25, 23, 18, 37, 46, 27, 52, 53, 67, 43, 50, 35, 26, 30, 23, 21, 15, 18, 28, 33, 12, 20, 46, 288, 54, 48, 51, 51, 75, 47, 31, 39, 55, 37, 27, 55, 49, 58, 59, 51, 30, 46, 58, 38, 46, 52, 186, 918, 40, 61, 51, 66, 46, 44, 57, 35, 61, 42, 45, 72, 46, 45, 33, 46, 45, 36, 32, 61, 56]},
{'total': 4952, 'pred_idx_mask': [75, 19, 34, 37, 23, 29, 25, 23, 30, 33, 46, 31, 231, 34, 36, 28, 37, 41, 45, 21, 24, 33, 22, 21, 34, 28, 97, 28, 32, 203, 218, 22, 48, 51, 27, 18, 28, 33, 22, 25, 39, 32, 17, 23, 71, 55, 52, 51, 61, 48, 55, 33, 47, 38, 48, 53, 37, 32, 57, 47, 51, 45, 33, 50, 35, 39, 81, 44, 454, 206, 37, 214, 50, 47, 55, 48, 44, 32, 57, 53, 42, 45, 48, 56, 32, 44, 54, 41, 35, 32, 60]},
               {'total': 4952,
                'pred_idx_masks': [102, 33, 25, 31, 26, 28, 62, 17, 33, 39, 24, 20, 82, 34, 45, 43, 35, 44, 25, 29, 38,
                                  41, 30, 20, 27, 20, 120, 12, 37, 162, 140, 28, 63, 48, 48, 27, 30, 71, 17, 84, 42, 17,
                                  18, 20, 33, 245, 64, 59, 56, 34, 42, 35, 30, 70, 48, 34, 34, 57, 41, 36, 52, 42, 80,
                                  45, 47, 41, 157, 34, 61, 222, 41, 164, 48, 24, 36, 72, 46, 118, 289, 24, 25, 50, 38,
                                  109, 22, 28, 36, 35, 46, 22, 43],
                'pred_idx_mask': [187, 44, 31, 38, 22, 26, 53, 32, 36, 42, 14, 9, 123, 33, 26, 51, 29, 37, 38, 25, 37, 32,
                             31, 28, 12, 24, 188, 15, 23, 98, 146, 23, 66, 42, 57, 20, 24, 48, 33, 59, 38, 24, 15, 19,
                             36, 162, 66, 46, 44, 41, 68, 34, 39, 56, 58, 32, 33, 56, 56, 42, 59, 31, 84, 44, 36, 30,
                             94, 32, 56, 267, 26, 206, 60, 41, 44, 56, 54, 125, 262, 39, 23, 53, 41, 61, 44, 29, 38, 52,
                             35, 24, 39]}])

for scores in all_scores:
    for i in range(len(scores["pred_idx_mask"])):
        scores['pred_idx_mask'][i] /= scores["total"]
x = []
for i in range(len(all_scores)):
    x.append([all_scores[i]["pred_idx_mask"][0]])

for j, scores in enumerate(all_scores):
    for i in range(1, len(scores["pred_idx_mask"])):
        val = x[j][-1] + scores["pred_idx_mask"][i]
        x[j].append(val)

x =  x[-4:]
ideal = (np.arange(90)/90)**0.5

# for i in range(len(x)):
#     plt.plot(x[i], label = f"{i}")

plt.plot(x[0], label="UNET")
plt.plot(x[1], label = "Simple CNN")
plt.plot(x[2], label = "Weighted UNET")
plt.plot(ideal, '--', label = "target plot")

plt.xlabel("Top n")
plt.ylabel("Accuracy")
plt.title("Cumulative top n accuracy plot.")
plt.legend()
plt.show()