mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

fix coords convert

Browse Source
This commit is contained in:
Francesco Gatti
2020-07-27 13:52:39 +02:00
parent b4015a8e26
commit 8c8341cc61
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/Yolo3Detection.cpp
+5 -5
View File
@@ -114,17 +114,17 @@ void Yolo3Detection::postprocess(const int bi, const bool mAP){
int y0 = (b.y-b.h/2.); int y0 = (b.y-b.h/2.);
int y1 = (b.y+b.h/2.); int y1 = (b.y+b.h/2.);
for(int c=0; c<classes; c++) {
if(dets[j].prob[c] >= confThreshold) {
int obj_class = c;
float prob = dets[j].prob[c];
// convert to image coords // convert to image coords
x0 = x_ratio*x0; x0 = x_ratio*x0;
x1 = x_ratio*x1; x1 = x_ratio*x1;
y0 = y_ratio*y0; y0 = y_ratio*y0;
y1 = y_ratio*y1; y1 = y_ratio*y1;
for(int c=0; c<classes; c++) {
if(dets[j].prob[c] >= confThreshold) {
int obj_class = c;
float prob = dets[j].prob[c];
tk::dnn::box res; tk::dnn::box res;
res.cl = obj_class; res.cl = obj_class;
res.prob = prob; res.prob = prob;