Commit 3e48a36e authored by Davis King's avatar Davis King

The loss returned by compute_loss_value_and_gradient() wasn't quite right. It

doesn't effect normal use, but it's still wrong and this change fixes it.
parent 3da61b25
......@@ -831,7 +831,7 @@ namespace dlib
// We are ignoring this box so we shouldn't have counted it in the
// loss in the first place. So we subtract out the loss values we
// added for it in the code above.
loss -= 1-out_data[idx];
loss -= options.loss_per_missed_target-out_data[idx];
g[idx] = 0;
std::cout << "Warning, ignoring object. We encountered a truth rectangle located at " << (*truth)[i].rect;
std::cout << " that is suppressed by non-max-suppression ";
......
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