派生自 Algorithm/baseDetector

Scheaven
2021-08-11 8e10c57c7e053d8789747cf1e2c5fa78f2f65cc7
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
@@ -42,6 +52,11 @@
{
    static std::string model_path;
    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