Skip to content

Commit fb74a0d

Browse files
committed
fix callback bug
1 parent d77e91e commit fb74a0d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

utils/callbacks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ def get_map_txt(self, image_id, image, class_names, map_out_path):
149149
top_conf = results[0][:, 4] * results[0][:, 5]
150150
top_boxes = results[0][:, :4]
151151

152-
top_100 = np.argsort(top_label)[::-1][:self.max_boxes]
152+
top_100 = np.argsort(top_conf)[::-1][:self.max_boxes]
153153
top_boxes = top_boxes[top_100]
154154
top_conf = top_conf[top_100]
155155
top_label = top_label[top_100]

0 commit comments

Comments
 (0)