This commit is contained in:
Glenn Jocher 2018-09-10 16:35:00 +02:00
parent 34144aabe3
commit ff04315f96
1 changed files with 1 additions and 1 deletions

View File

@ -99,7 +99,7 @@ def ap_per_class(tp, conf, pred_cls, target_cls):
tp, conf, pred_cls = tp[i], conf[i], pred_cls[i] tp, conf, pred_cls = tp[i], conf[i], pred_cls[i]
# Find unique classes # Find unique classes
unique_classes = np.unique(np.concatenate((pred_cls, target_cls), 0)) unique_classes = target_cls# np.unique(np.concatenate((pred_cls, target_cls), 0))
# Create Precision-Recall curve and compute AP for each class # Create Precision-Recall curve and compute AP for each class
ap = [] ap = []