"vscode:/vscode.git/clone" did not exist on "6c10d0809a199c78e9495d824a2298578ff846ac"
Commit 4b92804d authored by Davis King's avatar Davis King
Browse files

Use the box with bounding box regression applied to do NMS in the loss.

parent 0e721e5c
......@@ -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;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment