#include #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;