From 4b92804dc2fe6642a360af7e97adcbd7cf8c6dc7 Mon Sep 17 00:00:00 2001 From: Davis King Date: Tue, 1 Sep 2020 06:58:35 -0400 Subject: [PATCH] Use the box with bounding box regression applied to do NMS in the loss. --- dlib/dnn/loss.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dlib/dnn/loss.h b/dlib/dnn/loss.h index a2fb07908..7ac40a7b6 100644 --- a/dlib/dnn/loss.h +++ b/dlib/dnn/loss.h @@ -1487,7 +1487,7 @@ namespace dlib // The point of this loop is to fill out the truth_score_hits array. for (size_t i = 0; i < dets.size() && final_dets.size() < max_num_dets; ++i) { - if (overlaps_any_box_nms(final_dets, dets[i].rect)) + if (overlaps_any_box_nms(final_dets, dets[i].rect_bbr)) continue; const auto& det_label = options.detector_windows[dets[i].tensor_channel].label; @@ -1556,7 +1556,7 @@ namespace dlib // detections. for (unsigned long i = 0; i < dets.size() && final_dets.size() < max_num_dets; ++i) { - if (overlaps_any_box_nms(final_dets, dets[i].rect)) + if (overlaps_any_box_nms(final_dets, dets[i].rect_bbr)) continue; const auto& det_label = options.detector_windows[dets[i].tensor_channel].label;