Yolov3_tiny works on tensorRT

Signed-off-by: fbagni <gattinomicino>
2019-12-23 16:11:41 +01:00
parent f0ea2c027f
commit 7e7f480e2b
5 changed files with 20 additions and 32 deletions
@@ -24,8 +24,8 @@ Yolo::Yolo(Network *net, int classes, int num, std::string fname_weights, int n_
        readBinaryFile(fname_weights, n_masks, &mask_h, &mask_d, seek);
        seek += n_masks;
        readBinaryFile(fname_weights, n_masks*num*2, &bias_h, &bias_d, seek);
-        for(int i=0; i<n_masks*num*2; i++)
-            printf("%f\n", bias_h[i]);
+        //for(int i=0; i<n_masks*num*2; i++)
+            //printf("%f\n", bias_h[i]);
    }

    // init default classes name