//
|
// Created by Scheaven on 2020/3/24.
|
//
|
|
#ifndef INC_01_CPP_SORT_CONFIG_H
|
#define INC_01_CPP_SORT_CONFIG_H
|
#include "utils/time_util.h"
|
|
#include <opencv2/opencv.hpp>
|
#include <algorithm>
|
#include <assert.h>
|
#include <cmath>
|
#include <fstream>
|
#include <iomanip>
|
#include <sstream>
|
#include <sys/stat.h>
|
#include <time.h>
|
#include <cstdlib>
|
#include <iostream>
|
#include <string>
|
#include "argsParser.h"
|
#include "buffers.h"
|
#include "common.h"
|
// #include "parserOnnxConfig.h"
|
|
#include <map>
|
#include <chrono>
|
#include "logging.h"
|
#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
|
unsigned int obj_id; // class of object - from range [0, classes-1]
|
unsigned int track_id; // tracking id for video (0 - untracked, 1 - inf - tracked object)
|
unsigned int frames_counter; // counter of frames on which the object was detected
|
float x_3d, y_3d, z_3d; // center of object (in Meters) if ZED 3D Camera is used
|
};
|
|
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
|
|
#define CLASS_BUM 2
|
