fix callback bug

a04b460e · Bubbliiiing · 4821507c · a04b460e
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

utils/callbacks.py utils/callbacks.py +1 -1

未找到文件。
--- a/utils/callbacks.py
+++ b/utils/callbacks.py
@@ -149,7 +149,7 @@ class EvalCallback():
            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]