modularization/src/detection/laneATT_trt/para.hpp

#include <chrono>
#include "cuda_utils.h"
#include "logging.h"
#include "utils.h"

static const int INPUT_H = Yolo::INPUT_H;
static const int INPUT_W = Yolo::INPUT_W;
static const int IMG_H = Yolo::IMG_H;
static const int IMG_W = Yolo::IMG_W;
static const int CLASS_NUM = Yolo::CLASS_NUM;
//static const int OUTPUT_SIZE = Yolo::MAX_OUTPUT_BBOX_COUNT * sizeof(Yolo::Detection) / sizeof(float) + 1;  // we assume the yololayer outputs no more than MAX_OUTPUT_BBOX_COUNT boxes that conf >= 0.1
static const int OUTPUT_SIZE = 77000;
const char* INPUT_BLOB_NAME = "input";
const char* OUTPUT_DET_NAME = "split_for_trace_model_pred";
const char* OUTPUT_SEG_NAME = "seg";
const char* OUTPUT_LANE_NAME = "ll";
static Logger gLogger;