14 lines
360 B
C++
14 lines
360 B
C++
#pragma once
|
|
#include "types_model.hpp"
|
|
|
|
struct DetectBoxV10 {
|
|
int cls; // 0=锥 1=红灯 2=绿灯 3=行人
|
|
float conf;
|
|
float cx, cy, w, h;
|
|
};
|
|
|
|
bool model_v10_init(const char* path);
|
|
int model_v10_detect(const uint8* bgr, int w, int h, DetectBoxV10* boxes, int max, const float* thresh);
|
|
void model_v10_deinit();
|
|
bool model_v10_ready();
|