From 8fa9f44211e5b06a5e4c0859eb31af6a45492222 Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Tue, 12 Sep 2017 14:16:21 +0000 Subject: [PATCH] Improved speed of yolo_console_dll.cpp - 40 FPS on 4K using GeForce GTX 960 --- src/box.h | 1 + 1 files changed, 1 insertions(+), 0 deletions(-) diff --git a/src/box.h b/src/box.h index 9b57fb4..a5f8cee 100644 --- a/src/box.h +++ b/src/box.h @@ -14,6 +14,7 @@ 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); +void do_nms_sort(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