Skip to content

Commit

Permalink
Update detection.py
Browse files Browse the repository at this point in the history
  • Loading branch information
felixhjh authored Jul 22, 2022
1 parent 546e276 commit ce4e66b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion fastdeploy/vision/evaluation/detection.py
Original file line number Diff line number Diff line change
Expand Up @@ -175,4 +175,4 @@ def eval_detection(model,
ap50, ap = ap[:, 0], ap.mean(1) # AP@0.5, AP@0.5:0.95
mp, mr, map50, map = round(p.mean(), 3), round(r.mean(), 3), round(
ap50.mean(), 3), round(ap.mean(), 3)
return {"Precision": mp, "Recall": mr, "mAP@.5": map50, "mAP@.5:.95": map}
return {"Precision": mp, "Recall": mr, "mAP@.5": map50, "mAP@.5:.95": map}

0 comments on commit ce4e66b

Please sign in to comment.