111 lines
3.0 KiB
C
111 lines
3.0 KiB
C
|
#ifndef __RKNN_SDK_H__
|
|||
|
|
#define __RKNN_SDK_H__
|
|||
|
|
|
|||
|
|
#include <stdint.h>
|
|||
|
|
|
|||
|
|
// 返回错误码
|
|||
|
|
enum XErrorCode {
|
|||
|
|
RKNN_SUCCEED = 0,
|
|||
|
|
RKNN_NOT_INIT = -1, //SDK未初始化
|
|||
|
|
RKNN_INVAILD_IMGSIZE = -2, //输入图像尺寸不匹配模型输入要求
|
|||
|
|
RKNN_INVAILD_PARAM = -3, //参数无效
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
enum ObjCls {
|
|||
|
|
OBJ_Unknown = -1, //未知
|
|||
|
|
OBJ_MaoGouWan = 0, //猫碗/狗碗
|
|||
|
|
OBJ_BTYDiZuo = 1, //吧台椅底座
|
|||
|
|
OBJ_FSDiZuo = 2, //风扇底座
|
|||
|
|
OBJ_XiYiJi = 3, //洗衣机
|
|||
|
|
OBJ_BingXiang = 4, //冰箱
|
|||
|
|
OBJ_MaTong = 5, //马桶
|
|||
|
|
OBJ_TiZhongCheng = 6, //体重秤
|
|||
|
|
OBJ_DianXian = 7, //电线
|
|||
|
|
OBJ_DianShiJi = 8, //电视机
|
|||
|
|
OBJ_CanZhuo = 9, //餐桌
|
|||
|
|
OBJ_DiTan = 10, //地毯
|
|||
|
|
OBJ_MaBu = 11, //抹布
|
|||
|
|
OBJ_ChaJi = 12, //茶几
|
|||
|
|
OBJ_DianShiGui = 13, //电视柜
|
|||
|
|
OBJ_XieZi = 14, //拖鞋/鞋子
|
|||
|
|
OBJ_WaZi = 15, //袜子
|
|||
|
|
OBJ_YiGui = 16, //衣柜
|
|||
|
|
OBJ_Chuang = 17, //床
|
|||
|
|
OBJ_ShaFa = 18, //沙发
|
|||
|
|
OBJ_YiZi = 19, //椅子
|
|||
|
|
OBJ_Men = 20, //门
|
|||
|
|
OBJ_DLS = 21, //da li shi
|
|||
|
|
OBJ_DiBan = 22 //di ban
|
|||
|
|
//OBJ_Xiang = 21 //宠物粪便
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
|
|||
|
|
// 图像信息与buffer
|
|||
|
|
typedef struct _InputData {
|
|||
|
|
int shape[4]; // Num, Channel, Height, Width
|
|||
|
|
void* data; // img.data
|
|||
|
|
} InputData;
|
|||
|
|
|
|||
|
|
// 检测结果
|
|||
|
|
typedef struct _DetRet
|
|||
|
|
{
|
|||
|
|
int nID; //图像ID
|
|||
|
|
int nLabel; //物体类别 ObjCls nLabel; //物体类别
|
|||
|
|
float fConf; //置信度
|
|||
|
|
float fLeft; //检测框左上角比例坐标x_min -> int(x_min*width) 获取像素坐标
|
|||
|
|
float fTop; //检测框左上角比例坐标y_min -> int(y_min*height) 获取像素坐标
|
|||
|
|
float fRight; //检测框右下角比例坐标x_max -> int(x_max*width) 获取像素坐标
|
|||
|
|
float fBottom; //检测框右下角比例坐标y_max -> int(y_max*height) 获取像素坐标
|
|||
|
|
} DetRet;
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
#define RKNNSDK_API(retType) retType
|
|||
|
|
|
|||
|
|
///
|
|||
|
|
#ifdef __cplusplus
|
|||
|
|
extern "C" {
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
typedef uint64_t rknn_handle;
|
|||
|
|
|
|||
|
|
|
|||
|
|
/* RKNN_Create
|
|||
|
|
SDK加载模型，仅初始化一次;
|
|||
|
|
input:
|
|||
|
|
rknn_handle hd 句柄
|
|||
|
|
const char* model_path 模型路径
|
|||
|
|
return:
|
|||
|
|
int 错误码
|
|||
|
|
*/
|
|||
|
|
RKNNSDK_API(int) RKNN_Create(rknn_handle* hd, const char* model_path);
|
|||
|
|
|
|||
|
|
/* RKNN_ObjDet
|
|||
|
|
输入图像，获取检测结果和检测目标数量;
|
|||
|
|
input:
|
|||
|
|
rknn_handle hd 句柄
|
|||
|
|
const InputData* input_data 输入图像数据
|
|||
|
|
DetRet** det_ret 获取检测结果
|
|||
|
|
int* det_num 获取检测目标数量
|
|||
|
|
return:
|
|||
|
|
int 错误码
|
|||
|
|
*/
|
|||
|
|
RKNNSDK_API(int) RKNN_ObjDet(rknn_handle hd, const InputData* input_data, DetRet** det_ret, int* det_num);
|
|||
|
|
|
|||
|
|
/* RKNN_Release
|
|||
|
|
SDK资源释放，仅在程序退出前调用;
|
|||
|
|
input:
|
|||
|
|
rknn_handle hd 句柄
|
|||
|
|
return:
|
|||
|
|
int 错误码
|
|||
|
|
*/
|
|||
|
|
RKNNSDK_API(int) RKNN_Release(rknn_handle hd);
|
|||
|
|
|
|||
|
|
#ifdef __cplusplus
|
|||
|
|
}
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
#endif //__RKNN_SDK_H__
|