Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-12-21 09:32:47 -08:00
parent 083d482561
commit 587b7a8dd0
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
utils/utils.py
View File

@ -172,8 +172,8 @@ def ap_per_class(tp, conf, pred_cls, target_cls):
ap, p, r = np.zeros(s), np.zeros(s), np.zeros(s) ap, p, r = np.zeros(s), np.zeros(s), np.zeros(s)
for ci, c in enumerate(unique_classes): for ci, c in enumerate(unique_classes):
i = pred_cls == c i = pred_cls == c
n_gt = sum(target_cls == c) # Number of ground truth objects n_gt = (target_cls == c).sum() # Number of ground truth objects
n_p = sum(i) # Number of predicted objects n_p = i.sum() # Number of predicted objects
if n_p == 0 or n_gt == 0: if n_p == 0 or n_gt == 0:
continue continue