diff --git a/utils/callbacks.py b/utils/callbacks.py index 8ae0a04..918a527 100644 --- a/utils/callbacks.py +++ b/utils/callbacks.py @@ -144,7 +144,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]