From ce9d187fd294cca192a27f52719094e9df7b1b62 Mon Sep 17 00:00:00 2001 From: 孙天宇 <suntianyu@aiotlink.com> Date: 星期二, 12 七月 2022 16:28:48 +0800 Subject: [PATCH] 优化仓库 --- src/detecter_tools/detector.cpp | 11 +++++------ 1 files changed, 5 insertions(+), 6 deletions(-) diff --git a/src/detecter_tools/detector.cpp b/src/detecter_tools/detector.cpp index cf3e313..be5f86a 100644 --- a/src/detecter_tools/detector.cpp +++ b/src/detecter_tools/detector.cpp @@ -31,10 +31,10 @@ vec_ds_images.emplace_back(img, _vec_net_type[_config.net_type], _p_net->getInputH(), _p_net->getInputW()); } cv::Mat trtInput = blobFromDsImages(vec_ds_images, _p_net->getInputH(),_p_net->getInputW()); - timer.out("pre"); + timer.out("eve pre detect "); - _p_net->doInference(trtInput.data, vec_ds_images.size()); timer.reset(); + _p_net->doInference(trtInput.data, vec_ds_images.size()); for (uint32_t i = 0; i < vec_ds_images.size(); ++i) { auto curImage = vec_ds_images.at(i); @@ -65,9 +65,8 @@ } vec_batch_result[i] = vec_result; } - timer.out("post"); + timer.out("eve pre detect post"); DEBUG("--detect over--" ); - } void Detector::set_gpu_id(const int id) @@ -96,8 +95,8 @@ void Detector::build_net() { if(_config.net_type == COMMON) - _p_net = std::unique_ptr<Detecter>{new Detecter(_info,_infer_param,0)}; - else{ _p_net = std::unique_ptr<Detecter>{new Detecter(_info,_infer_param,1)}; + else{ + _p_net = std::unique_ptr<Detecter>{new Detecter(_info,_infer_param,2)}; } } -- Gitblit v1.8.0