From 8bcdee86585f496afe1a8a38d608ea0504a11243 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Tue, 01 Sep 2015 18:22:03 +0000 Subject: [PATCH] Some bug fixes, random stuff --- src/box.h | 4 ++++ 1 files changed, 4 insertions(+), 0 deletions(-) diff --git a/src/box.h b/src/box.h index e3831d8..f7ef36f 100644 --- a/src/box.h +++ b/src/box.h @@ -10,6 +10,10 @@ } 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 total, int classes, float thresh); +box decode_box(box b, box anchor); +box encode_box(box b, box anchor); #endif -- Gitblit v1.10.0