diff --git a/demo/demo/map.cpp b/demo/demo/map.cpp index e32c302..a5d7efa 100644 --- a/demo/demo/map.cpp +++ b/demo/demo/map.cpp @@ -66,9 +66,24 @@ int main(int argc, char *argv[]) IoU_thresh, conf_thresh, verbose); std::ofstream times, memory; + std::string name; if(write_res_on_file) { - times.open("times"+std::string(net)+".csv"); + std::string str = net; + name = net; + std::string delim = "/"; + std::size_t current, previous = 0; + current = str.find(delim); + if (current != std::string::npos) { + while (current != std::string::npos) { + name = str.substr(previous, current - previous); + previous = current + 1; + current = str.find(delim, previous); + } + name = str.substr(previous, current - previous); + } + std::cout<<"name: "<