updates

2020-03-11 21:30:47 -07:00
parent 2d32423461
commit 6ca8277de2
2 changed files with 2 additions and 2 deletions
@@ -188,7 +188,7 @@ def ap_per_class(tp, conf, pred_cls, target_cls):
    unique_classes = np.unique(target_cls)

    # Create Precision-Recall curve and compute AP for each class
-    pr_score = 0.5  # score to evaluate P and R https://github.com/ultralytics/yolov3/issues/898
+    pr_score = 0.1  # score to evaluate P and R https://github.com/ultralytics/yolov3/issues/898
    s = [len(unique_classes), tp.shape[1]]  # number class, number iou thresholds (i.e. 10 for mAP0.5...0.95)
    ap, p, r = np.zeros(s), np.zeros(s), np.zeros(s)
    for ci, c in enumerate(unique_classes):