From c53e03348c65462bcba33f6352087dd6b9268e8f Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Wed, 16 Sep 2015 21:12:10 +0000 Subject: [PATCH] yolo working w/ regions --- src/box.h | 3 ++- 1 files changed, 2 insertions(+), 1 deletions(-) diff --git a/src/box.h b/src/box.h index e45dd89..f7ef36f 100644 --- a/src/box.h +++ b/src/box.h @@ -10,8 +10,9 @@ } dbox; float box_iou(box a, box b); +float box_rmse(box a, box b); dbox diou(box a, box b); -void do_nms(box *boxes, float **probs, int num_boxes, int classes, float thresh); +void do_nms(box *boxes, float **probs, int total, int classes, float thresh); box decode_box(box b, box anchor); box encode_box(box b, box anchor); -- Gitblit v1.10.0