diff --git a/dlib/dnn/loss.h b/dlib/dnn/loss.h index 86fc00c2d..bce1d3473 100644 --- a/dlib/dnn/loss.h +++ b/dlib/dnn/loss.h @@ -881,10 +881,26 @@ namespace dlib deserialize(item.options, in); } - friend std::ostream& operator<<(std::ostream& out, const loss_mmod_& ) + friend std::ostream& operator<<(std::ostream& out, const loss_mmod_& item) { - // TODO, add options fields - out << "loss_mmod"; + out << "loss_mmod\t ("; + + out << "detector_windows:("; + auto& opts = item.options; + for (size_t i = 0; i < opts.detector_windows.size(); ++i) + { + out << opts.detector_windows[i].width << "x" << opts.detector_windows[i].height; + if (i+1 < opts.detector_windows.size()) + out << ","; + } + out << ")"; + out << ", loss per FA:" << opts.loss_per_false_alarm; + out << ", loss per miss:" << opts.loss_per_missed_target; + out << ", truth match IOU thresh:" << opts.truth_match_iou_threshold; + out << ", overlaps_nms:("<