fee7cbd0fe
* yolov8 * Update README.md * Update README.md * Update main.cpp * Delete yolov8/output directory * Delete yolov8/output directory
15 lines
465 B
C
15 lines
465 B
C
#define USE_FP16
|
|
//#define USE_INT8
|
|
|
|
const static char *kInputTensorName = "images";
|
|
const static char *kOutputTensorName = "output";
|
|
const static int kNumClass = 80;
|
|
const static int kBatchSize = 1;
|
|
const static int kGpuId = 0;
|
|
const static int kInputH = 640;
|
|
const static int kInputW = 640;
|
|
const static float kNmsThresh = 0.45f;
|
|
const static float kConfThresh = 0.5f;
|
|
const static int kMaxInputImageSize = 3000 * 3000;
|
|
const static int kMaxNumOutputBbox = 1000;
|