从内存加载明文模型 - YoloLoadModelMemoryEx
函数简介
从内存加载明文主模型,可选 NCNN .param 与类别名表。
接口名称
YoloLoadModelMemoryEx
DLL 调用
long YoloLoadModelMemoryEx(long ola, long modelData, int modelSize, long ncnnParamData, int ncnnParamSize, long labelData, int labelSize, int modelType, int inferenceType, int inferenceDevice);
参数说明
| 参数名 | 类型 | 说明 |
|---|---|---|
| ola | 长整数型 | OLAPlug 对象指针,由 CreateCOLAPlugInterFace 生成。 |
| modelData | 长整数型 | 主模型内存地址 |
| modelSize | 整数型 | 主模型字节数 |
| ncnnParamData | 长整数型 | NCNN .param 地址,无则 0 |
| ncnnParamSize | 整数型 | NCNN .param 长度 |
| labelData | 长整数型 | 类别名表地址,无则 0 |
| labelSize | 整数型 | 类别名表字节长度 |
| modelType | 整数型 | 模型类型 |
| inferenceType | 整数型 | 推理类型 |
| inferenceDevice | 整数型 | inferenceDevice:-1 表示 CPU;0 及以上表示 GPU 索引。GPU 不可用时可能自动回退 CPU(以实际 ExecutionProvider 为准)。 |
返回值
长整数型:模型句柄,失败返回 0。
注意事项
- 需要插件已开通 Yolos 模块权限。
| 值 | 含义 |
|---|---|
| 0 | TensorRT Engine(.engine) |
| 1 | ONNX(.onnx) |
| 2 | NCNN(.bin + .param 双文件) |
| 值 | 含义 |
|---|---|
| 0 | Detect 目标检测 |
| 1 | Classify 图像分类 |
| 2 | Segment 实例分割 |
| 3 | Pose 姿态估计 |
| 4 | Obb 旋转框检测 |
inferenceDevice:-1表示 CPU;0及以上表示 GPU 索引。GPU 不可用时可能自动回退 CPU(以实际 ExecutionProvider 为准)。- 缓冲区不得为加密包格式。
