Fixed boxes to float, add conf thresh as param

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-08-05 19:55:10 +02:00
parent a5d2d4792a
commit f778e1aa99
11 changed files with 22 additions and 15 deletions
@@ -84,7 +84,7 @@ class DetectionNN {
         * @param n_batches maximum number of batches to use in inference
         * @return true if everything is correct, false otherwise.
         */
-        virtual bool init(const std::string& tensor_path, const int n_classes=80, const int n_batches=1) = 0;
+        virtual bool init(const std::string& tensor_path, const int n_classes=80, const int n_batches=1, const float conf_thresh=0.3) = 0;
        
        /**
         * This method performs the whole detection of the NN.