class in box

2017-08-22 06:37:43 -07:00
parent 37b050a9c8
commit 0119b31455
3 changed files with 5 additions and 2 deletions
@@ -256,6 +256,7 @@ void RegionInterpret::interpretData(dnnType *data_h) {
    }
    //

+    res_boxes_n = 0;
    //print results
    for(int i = 0; i < tot; ++i){
        int cl = max_index(probs[i], classes);
@@ -273,6 +274,7 @@ void RegionInterpret::interpretData(dnnType *data_h) {
            b.y = y;
            b.h = h;
            b.w = w;
+            b.cl = cl;
            res_boxes[res_boxes_n] = b;
            res_boxes_n++;
        }