Add json detection creation for codalab check

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-05-11 11:57:58 +02:00
parent adb5a693cd
commit 533bb48789
17 changed files with 2119 additions and 34 deletions
@@ -535,6 +535,7 @@ struct box {
    int cl;
    float x, y, w, h;
    float prob;
+    std::vector<float> probs;

    void print() 
    {
@@ -581,7 +582,7 @@ public:

    dnnType *predictions;

-    static const int MAX_DETECTIONS = 2048;
+    static const int MAX_DETECTIONS = 8192;
    static Yolo::detection *allocateDetections(int nboxes, int classes);
    static void             mergeDetections(Yolo::detection *dets, int ndets, int classes);
 };