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 | 7 +++----
1 files changed, 3 insertions(+), 4 deletions(-)
diff --git a/src/detecter_tools/detector.cpp b/src/detecter_tools/detector.cpp
index 1a3d1aa..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)
--
Gitblit v1.8.0