Skip to content

Commit

Permalink
fix callback bug
Browse files Browse the repository at this point in the history
  • Loading branch information
bubbliiiing committed Jul 16, 2022
1 parent d77e91e commit fb74a0d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion utils/callbacks.py
Original file line number Diff line number Diff line change
Expand Up @@ -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]
Expand Down

0 comments on commit fb74a0d

Please sign in to comment.