fix callback bug

bubbliiiing · Jul 16, 2022 · 79aa7c7 · 79aa7c7
1 parent 6d7a682
commit 79aa7c7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/callbacks.py b/utils/callbacks.py
@@ -149,7 +149,7 @@ def get_map_txt(self, image_id, image, class_names, map_out_path):
             top_conf    = results[0][:, 4] * results[0][:, 5]
             top_boxes   = results[0][:, :4]
 
-        top_100     = np.argsort(top_label)[::-1][:self.max_boxes]
+        top_100     = np.argsort(top_conf)[::-1][:self.max_boxes]
         top_boxes   = top_boxes[top_100]
         top_conf    = top_conf[top_100]
         top_label   = top_label[top_100]