| 123 | } |
| 124 | |
| 125 | cv::Mat NeuralNetworkClassifier::draw_result(const Result& res) const |
| 126 | { |
| 127 | cv::Mat image_draw = draw_roi(); |
| 128 | cv::Point pt(res.box.x + res.box.width + 5, res.box.y + 20); |
| 129 | |
| 130 | for (size_t i = 0; i != res.raw.size(); ++i) { |
| 131 | const auto color = i == res.cls_index ? cv::Scalar(0, 0, 255) : cv::Scalar(255, 0, 0); |
| 132 | std::string text = std::format("{} {}: prob {:.3f}, raw {:.3f}", i, param_.labels[i], res.probs[i], res.raw[i]); |
| 133 | cv::putText(image_draw, text, pt, cv::FONT_HERSHEY_PLAIN, 1.2, color, 1); |
| 134 | pt.y += 20; |
| 135 | } |
| 136 | |
| 137 | return image_draw; |
| 138 | } |
| 139 | |
| 140 | void NeuralNetworkClassifier::sort_(ResultsVec& results) const |
| 141 | { |