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