From c7af21d4a444b41293d0372f6b45ae77c80bd0fb Mon Sep 17 00:00:00 2001
From: Scheaven <xuepengqiang>
Date: 星期三, 08 九月 2021 18:02:06 +0800
Subject: [PATCH] update
---
src/detecter_tools/detector.cpp | 13 ++++++-------
1 files changed, 6 insertions(+), 7 deletions(-)
diff --git a/src/detecter_tools/detector.cpp b/src/detecter_tools/detector.cpp
index 78e136f..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)
@@ -95,9 +94,9 @@
void Detector::build_net()
{
- if(_config.net_type == SMALL)
- _p_net = std::unique_ptr<Detecter>{new Detecter(_info,_infer_param,2)};
- else{
+ if(_config.net_type == COMMON)
_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