diff --git a/demo/demo/demo3D.cpp b/demo/demo/demo3D.cpp index 609a21a..4286faa 100644 --- a/demo/demo/demo3D.cpp +++ b/demo/demo/demo3D.cpp @@ -105,11 +105,24 @@ int main(int argc, char *argv[]) { std::cout<<"detection end\n"; double mean = 0; + std::cout<pre_stats.begin(), detNN->pre_stats.end())<<" ms\n"; + std::cout<<"Max: "<<*std::max_element(detNN->pre_stats.begin(), detNN->pre_stats.end())<<" ms\n"; + for(int i=0; ipre_stats.size(); i++) mean += detNN->pre_stats[i]; mean /= detNN->pre_stats.size(); + std::cout<<"Avg: "<stats.begin(), detNN->stats.end())<<" ms\n"; std::cout<<"Max: "<<*std::max_element(detNN->stats.begin(), detNN->stats.end())<<" ms\n"; for(int i=0; istats.size(); i++) mean += detNN->stats[i]; mean /= detNN->stats.size(); std::cout<<"Avg: "<post_stats.begin(), detNN->post_stats.end())<<" ms\n"; + std::cout<<"Max: "<<*std::max_element(detNN->post_stats.begin(), detNN->post_stats.end())<<" ms\n"; + for(int i=0; ipost_stats.size(); i++) mean += detNN->post_stats[i]; mean /= detNN->post_stats.size(); + std::cout<<"Avg: "< detected; /*bounding boxes in output*/ - std::vector stats; /*keeps track of inference times (ms)*/ + std::vector pre_stats, stats, post_stats, visual_stats; /*keeps track of inference times (ms)*/ std::vector classesNames; DetectionNN3D() {}; @@ -107,6 +107,7 @@ class DetectionNN3D { TKDNN_TSTART preprocess(frame); TKDNN_TSTOP + pre_stats.push_back(t_ns); if(save_times) *times<