Fix print on times.csv

Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-03-12 20:18:02 +01:00
parent 98770ee0c5
commit 5c347b1e11
3 changed files with 12 additions and 10 deletions
@@ -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;