Merge branch 'viniciusarruda-patch-1'

zhanghhong · Oct 12, 2018 · 29a5277 · 29a5277
2 parents 847326b + 5c4d59d
commit 29a5277
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -407,7 +407,7 @@ def draw_plot_func(dictionary, n_classes, window_title, plot_title, x_label, out
         bounding_boxes.append({"confidence":confidence, "file_id":file_id, "bbox":bbox})
         #print(bounding_boxes)
   # sort predictions by decreasing confidence
-  bounding_boxes.sort(key=lambda x:x['confidence'], reverse=True)
+  bounding_boxes.sort(key=lambda x:float(x['confidence']), reverse=True)
   with open(tmp_files_path + "/" + class_name + "_predictions.json", 'w') as outfile:
     json.dump(bounding_boxes, outfile)