From ce9d187fd294cca192a27f52719094e9df7b1b62 Mon Sep 17 00:00:00 2001
From: 孙天宇 <suntianyu@aiotlink.com>
Date: 星期二, 12 七月 2022 16:28:48 +0800
Subject: [PATCH] 优化仓库
---
src/config.h | 19 ++++++++++++++++++-
1 files changed, 18 insertions(+), 1 deletions(-)
diff --git a/src/config.h b/src/config.h
index f495e58..9673e66 100644
--- a/src/config.h
+++ b/src/config.h
@@ -4,7 +4,7 @@
#ifndef INC_01_CPP_SORT_CONFIG_H
#define INC_01_CPP_SORT_CONFIG_H
-#include "utils/timer_utils.hpp"
+#include "utils/time_util.h"
#include <opencv2/opencv.hpp>
#include <algorithm>
@@ -29,6 +29,16 @@
#include "cuda_runtime_api.h"
#include "NvInfer.h"
+const int nn_budget=100;
+const float max_cosine_distance=0.7;
+
+#define NN_BUDGET 100
+#define MAX_COSINE_DISTANCE 0.7
+#define MAX_IOU_DISTANCE 0.9
+#define MAX_AGE 450
+#define MAX_OUT_TIME 30.0
+#define N_INIT 5
+
struct bbox_t {
unsigned int x, y, w, h; // (x,y) - top-left corner, (w, h) - width & height of bounded box
float prob; // confidence - probability that the object was found correctly
@@ -41,7 +51,14 @@
typedef struct m_staticStruct
{
static std::string model_path;
+ static std::string model_cfg;
+ static std::string model_wts;
static int type;
+ static bool isTrack;
+ static int max_cam_num;
+ static int wander_time;
+ static float mv_velocity;
+ static float fall_rate;
}M_STATICSTRUCT, *P_STATICSTRUCT;
#endif //INC_01_CPP_SORT_CONFIG_H
--
Gitblit v1.8.0