Fix print on times.csv

Signed-off-by: xavier <micaelaverucchi@gmail.com>
This commit is contained in:
xavier
2020-03-12 20:18:02 +01:00
parent d0d462c015
commit db10ebeb34
3 changed files with 12 additions and 10 deletions
+2 -2
View File
@@ -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
conf_thresh : 0.3 #threshold on the condifence of the bbox
verbose : false #print on screen information
+9 -7
View File
@@ -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;
+1 -1
View File
@@ -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