优化仓库

孙天宇

2022-07-12 ce9d187fd294cca192a27f52719094e9df7b1b62

 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