Fix print on times.csv
Signed-off-by: xavier <micaelaverucchi@gmail.com>
This commit is contained in:
+2
-2
@@ -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
@@ -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;
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user