updates
This commit is contained in:
parent
34144aabe3
commit
ff04315f96
|
@ -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 = []
|
||||||
|
|
Loading…
Reference in New Issue