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 8c8341cc61
commit 498b163f36
11 changed files with 22 additions and 15 deletions
@@ -24,7 +24,7 @@ public:
    Yolo3Detection() {};
    ~Yolo3Detection() {}; 

-    bool init(const std::string& tensor_path, const int n_classes=80, const int n_batches=1);
+    bool init(const std::string& tensor_path, const int n_classes=80, const int n_batches=1, const float conf_thresh=0.3);
    void preprocess(cv::Mat &frame, const int bi=0);
    void postprocess(const int bi=0,const bool mAP=false);
 };