// // Created by Scheaven on 2020/3/24. // #ifndef INC_01_CPP_SORT_CONFIG_H #define INC_01_CPP_SORT_CONFIG_H #include #include "opencv2/core/core.hpp" #include "opencv2/highgui/highgui.hpp" #include #include #include #include #include #include #include #include #include #include "time.h" #include "stdio.h" #include "stdlib.h" #include #include //#define TRACHER_MODEL "../model_dump/tracker.pb" // 追踪模型 //#define COCO_FILE "../cfg/coco.names" //#define YOLO3_CFG "../cfg/yolov4.cfg" //#define YOLO3_WEIGHTS "../model_dump/yolov4.weights" //#define DETECT_GPU_ID 0 //检测模型的GPU //#define TRACHER_GPU_ID "/gpu:0" //追踪模型的GPU typedef unsigned char BYTE; typedef int (*fn)(void* args, const char* sender_chan, const int sender_len, const char *recver, const int recver_len, const char *data, const int len); const int nn_budget=100; const float max_cosine_distance=0.2; #define NN_BUDGET 100 // 保存特征的长度,,如果发生遮挡,会进入旧的特征进行匹配 #define MAX_COSINE_DISTANCE 0.2 //相似的最大COS距离 #define MAX_IOU_DISTANCE 0.7 // 匹配的最大IOU重合度 #define MAX_AGE 150 // 遮挡等待帧数 #define MAX_OUT_TIME 20.0 // 遮挡等待时间 #define N_INIT 0 #define MODEL_PATH "../model_dump/reid_model.pt" #define OUTPUT_DIR "./logs" #define GPU_ID 0 //gpu id 0代表第一块可见gpu #define REID_DISTANCE_THRESHOLD 95000000 //特征库 #define HUMAN_CONFIG_THRESHOLD 0.8 #define REID_MATCH_TIME_THRESHOLD 10000 // 重识别轨迹重复的强制匹配时长 #define GLOBALTRACKING false #endif //INC_01_CPP_SORT_CONFIG_H