Skip to content

GitLab

Commit 0e721e5c authored by Davis King's avatar Davis King
Browse files

Fix bug in bounding box regression loss.

parent c9809e06
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
dlib/dnn/loss.h
View file @ 0e721e5c
...@@ -1582,9 +1582,9 @@ namespace dlib ...@@ -1582,9 +1582,9 @@ namespace dlib
double dw = out_data[dets[i].tensor_offset_dw]; double dw = out_data[dets[i].tensor_offset_dw];
double dh = out_data[dets[i].tensor_offset_dh]; double dh = out_data[dets[i].tensor_offset_dh];
dpoint p = dcenter(dets[i].rect_bbr); dpoint p = dcenter(dets[i].rect);
double w = dets[i].rect_bbr.width()-1; double w = dets[i].rect.width()-1;
double h = dets[i].rect_bbr.height()-1; double h = dets[i].rect.height()-1;
drectangle truth_box = (*truth)[hittruth.second].rect; drectangle truth_box = (*truth)[hittruth.second].rect;
dpoint p_truth = dcenter(truth_box); dpoint p_truth = dcenter(truth_box);
......
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