Fix print on times.csv
Signed-off-by: xavier <micaelaverucchi@gmail.com>
This commit is contained in:
+1
-1
@@ -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_levels : 10 #number of IoU step for the AP
|
||||||
map_step : 0.05 #step of IoU
|
map_step : 0.05 #step of IoU
|
||||||
IoU_thresh : 0.5 #starting IoU threshold
|
IoU_thresh : 0.5 #starting IoU threshold
|
||||||
conf_thresh : 0.0 #threshold on the condifence of the bbox
|
conf_thresh : 0.3 #threshold on the condifence of the bbox
|
||||||
verbose : false #print on screen information
|
verbose : false #print on screen information
|
||||||
+9
-7
@@ -37,13 +37,6 @@ int main(int argc, char *argv[])
|
|||||||
bool write_res_on_file = true;
|
bool write_res_on_file = true;
|
||||||
int n_images = 5000;
|
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)
|
if(argc > 1)
|
||||||
net = argv[1];
|
net = argv[1];
|
||||||
if(argc > 2)
|
if(argc > 2)
|
||||||
@@ -53,6 +46,15 @@ int main(int argc, char *argv[])
|
|||||||
if(argc > 4)
|
if(argc > 4)
|
||||||
config_filename = argv[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::Yolo3Detection yolo;
|
||||||
tk::dnn::CenternetDetection cnet;
|
tk::dnn::CenternetDetection cnet;
|
||||||
|
|
||||||
|
|||||||
@@ -97,7 +97,7 @@ class CenternetDetection {
|
|||||||
int classes = 80;
|
int classes = 80;
|
||||||
int num = 0;
|
int num = 0;
|
||||||
int n_masks = 0;
|
int n_masks = 0;
|
||||||
float thresh = 0.3;
|
float thresh = 0.0;
|
||||||
cv::Scalar colors[256];
|
cv::Scalar colors[256];
|
||||||
|
|
||||||
// this is filled with results
|
// this is filled with results
|
||||||
|
|||||||
Reference in New Issue
Block a user