detect demo with mAP

2017-08-29 16:48:18 +00:00
parent ab45c24efc
commit b2d6dcd207
4 changed files with 225 additions and 16 deletions
@@ -276,6 +276,7 @@ public:
 struct box {
    int cl;
    float x, y, w, h;
+    float prob;
 };
 struct sortable_bbox {
    int index;
@@ -323,8 +324,10 @@ public:
                            float **probs, box *boxes, int only_objectness, 
                            int *map, float tree_thresh, int relative); 
    void correct_region_boxes(box *boxes, int n, int w, int h, int netw, int neth, int relative);
-    void interpretData(dnnType *data_h);
+    void interpretData(dnnType *data_h, int imageW = 0, int imageH = 0);
    void showImageResult(dnnType *input_h);
+
+    static float box_iou(box a, box b);
 };

 }