From 453436d9185a2367cf30f861506725ae0304b284 Mon Sep 17 00:00:00 2001
From: Scheaven <xuepengqiang>
Date: 星期三, 28 七月 2021 10:37:37 +0800
Subject: [PATCH] test code
---
src/core/ari_manager.cpp | 11 ++++++-----
demo.cpp | 2 +-
src/utils/draw_util.h | 3 +++
src/core/ari_manager.h | 1 +
src/utils/draw_util.cpp | 3 ++-
5 files changed, 13 insertions(+), 7 deletions(-)
diff --git a/demo.cpp b/demo.cpp
index b5145d6..6897241 100644
--- a/demo.cpp
+++ b/demo.cpp
@@ -48,7 +48,7 @@
for (int i=0; i<t_result->count; i++)
{
cout << "confidence:" << t_result->targets[i].confidence << endl;
- draw_SDK_result(frame, t_result->targets[i]);
+ // draw_SDK_result(frame, t_result->targets[i]);
}
}else{
cout << "------------over--" << endl;
diff --git a/src/core/ari_manager.cpp b/src/core/ari_manager.cpp
index 00ee3ef..857f820 100644
--- a/src/core/ari_manager.cpp
+++ b/src/core/ari_manager.cpp
@@ -60,12 +60,12 @@
this->detector->detect(batch_img, batch_res);
t_result.targets = (Target*)malloc(sizeof(Target)*batch_res[0].size());
- // 灏嗙畻娉曠粨鏋滆浆鍖栦负鏍囧噯鐨勬牸寮忥紙浠ョ洰鏍囨娴嬩负渚嬶級
+ // 灏嗙畻娉曠粨鏋滆浆鍖栦负鏍囧噯鐨勬牸寮忥紙浠ョ洰鏍囨娴嬩负渚嬶級""
int w_count = 0;
for (const auto &result_box:batch_res[0])
{
- // if(result_box.id == 0)
- // {
+ if(result_box.id == 1)
+ {
Target target;
init_target(&target);
@@ -91,10 +91,11 @@
t_result.targets[w_count] = target;
w_count ++;
- // }
+ }
}
+ std::cout << "eve batch_res size:: "<< batch_res[0].size() << " w_count: " << w_count <<std::endl;
t_result.count = w_count;
- draw_SDK_result(cam_id, frame, t_result); //澶氱嚎绋嬫棤娉曡皟鐢ㄧ粯鍥�
+ draw_SDK_result(cam_id, frame, t_result); //澶氱嚎绋嬫棤娉曡皟鐢ㄧ粯鍥�
}
void AriManager::init_target(Target *t){
diff --git a/src/core/ari_manager.h b/src/core/ari_manager.h
index e908f4f..9bec3f9 100644
--- a/src/core/ari_manager.h
+++ b/src/core/ari_manager.h
@@ -7,6 +7,7 @@
#include "std_target.h"
#include "config.h"
#include "detector.h"
+#include "../utils/draw_util.h"
using namespace std;
using namespace cv;
diff --git a/src/utils/draw_util.cpp b/src/utils/draw_util.cpp
index 4268f9f..5e2c5a7 100644
--- a/src/utils/draw_util.cpp
+++ b/src/utils/draw_util.cpp
@@ -80,7 +80,8 @@
cv::waitKey(0);
}
-cv::VideoWriter writer("/opt/vasystem/valog/01_Scheanve/01_baseDetector.avi", cv::VideoWriter::fourcc('M','J','P','G'),24, cv::Size(800,500), true);
+cv::VideoWriter writer("./01_baseDetector.avi", cv::VideoWriter::fourcc('M','J','P','G'),24, cv::Size(800,500), true);
+// cv::VideoWriter writer("/opt/vasystem/valog/01_Scheanve/01_baseDetector.avi", cv::VideoWriter::fourcc('M','J','P','G'),24, cv::Size(800,500), true);
void draw_SDK_result(const int cam_id, cv::Mat mat_img, TResult& t_result)
{
diff --git a/src/utils/draw_util.h b/src/utils/draw_util.h
index fa12331..8a916b1 100644
--- a/src/utils/draw_util.h
+++ b/src/utils/draw_util.h
@@ -12,5 +12,8 @@
void create_foldert(const char *path);
void draw_SDK_result(cv::Mat mat_img, Target& target);
+void draw_SDK_result(const int cam_id, cv::Mat mat_img, TResult& t_result);
+
+
#endif //DRAW_UTIL_H
--
Gitblit v1.8.0