mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Small fixes

Browse Source
This commit is contained in:
AlexeyAB
2017-08-07 19:13:58 +03:00
parent 0a8391e816
commit 4579b589ba
2 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/yolo_v2_class.hpp
View File

@ -64,7 +64,7 @@ public:
cv::Mat det_mat;
cv::resize(mat, det_mat, cv::Size(get_net_width(), get_net_height()));
auto image_ptr = mat_to_image(det_mat);
auto detection_boxes = detect(*image_ptr, thresh);
auto detection_boxes = detect(*image_ptr, thresh, use_mean);
float wk = (float)mat.cols / det_mat.cols, hk = (float)mat.rows / det_mat.rows;
for (auto &i : detection_boxes) i.x*=wk, i.w*= wk, i.y*=hk, i.h*=hk;
return detection_boxes;