pierreguillou commited on
Commit
d02ac89
1 Parent(s): 7e04d24

Update files/functions.py

Browse files
Files changed (1) hide show
  1. files/functions.py +1 -0
files/functions.py CHANGED
@@ -663,6 +663,7 @@ def predictions_line_level(dataset, outputs, images_ids_list, chunk_ids, input_i
663
  probs_label = list()
664
  for probs_list in probs:
665
  prob_label = reduce(lambda x, y: x*y, probs_list)
 
666
  probs_label.append(prob_label)
667
  max_value = max(probs_label)
668
  max_index = probs_label.index(max_value)
 
663
  probs_label = list()
664
  for probs_list in probs:
665
  prob_label = reduce(lambda x, y: x*y, probs_list)
666
+ prob_label = prob_label**(1./(len(probs_list))) # normalization
667
  probs_label.append(prob_label)
668
  max_value = max(probs_label)
669
  max_index = probs_label.index(max_value)