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
@@ -3,11 +3,12 @@

 namespace tk { namespace dnn {

-bool CenternetDetection::init(const std::string& tensor_path, const int n_classes, const int n_batches){
+bool CenternetDetection::init(const std::string& tensor_path, const int n_classes, const int n_batches, const float conf_thresh){
    std::cout<<(tensor_path).c_str()<<"\n";
    netRT = new tk::dnn::NetworkRT(NULL, (tensor_path).c_str() );
    classes = n_classes;
    nBatches = n_batches;
+    confThreshold = conf_thresh;

    dim = netRT->input_dim;