From db10ebeb34436870b5b910a41b5e84f94d8fa6a3 Mon Sep 17 00:00:00 2001
From: xavier <micaelaverucchi@gmail.com>
Date: Thu, 12 Mar 2020 20:18:02 +0100
Subject: [PATCH] Fix print on times.csv

Signed-off-by: xavier <micaelaverucchi@gmail.com>
---
 demo/config.yaml                   |  4 ++--
 demo/demo/map.cpp                  | 16 +++++++++-------
 include/tkDNN/CenternetDetection.h |  2 +-
 3 files changed, 12 insertions(+), 10 deletions(-)

diff --git a/demo/config.yaml b/demo/config.yaml
index 3c2d188..e86bf2b 100644
--- a/demo/config.yaml
+++ b/demo/config.yaml
@@ -3,5 +3,5 @@ map_points  : 101   #number of recall points (0 for all, 101 for COCO, 11 Pascal
 map_levels  : 10    #number of IoU step for the AP
 map_step    : 0.05  #step of IoU
 IoU_thresh  : 0.5   #starting IoU threshold
-conf_thresh : 0.0   #threshold on the condifence of the bbox
-verbose     : false #print on screen information
\ No newline at end of file
+conf_thresh : 0.3   #threshold on the condifence of the bbox
+verbose     : false #print on screen information
diff --git a/demo/demo/map.cpp b/demo/demo/map.cpp
index ac2b96a..7e24141 100644
--- a/demo/demo/map.cpp
+++ b/demo/demo/map.cpp
@@ -37,13 +37,6 @@ int main(int argc, char *argv[])
     bool write_res_on_file = true;
     int n_images = 5000;
 
-    std::ofstream times;
-    if(write_res_on_file)
-    {
-	times.open ("times.csv", std::ios_base::app);
-	times<<net<<";";
-    }
-    
     if(argc > 1)
         net = argv[1]; 
     if(argc > 2)
@@ -53,6 +46,15 @@ int main(int argc, char *argv[])
     if(argc > 4)
         config_filename = argv[4]; 
 
+
+    std::ofstream times;
+    if(write_res_on_file)
+    {
+        times.open ("times.csv", std::ios_base::app);
+        times<<net<<";";
+    }
+
+
     tk::dnn::Yolo3Detection yolo;
     tk::dnn::CenternetDetection cnet;
     
diff --git a/include/tkDNN/CenternetDetection.h b/include/tkDNN/CenternetDetection.h
index 7fd1c17..a2a656f 100644
--- a/include/tkDNN/CenternetDetection.h
+++ b/include/tkDNN/CenternetDetection.h
@@ -97,7 +97,7 @@ class CenternetDetection {
         int classes = 80;
         int num = 0;
         int n_masks = 0;
-        float thresh = 0.3;
+        float thresh = 0.0;
         cv::Scalar colors[256];
 
         // this is filled with results