主要功能
本 SDK 包含 SSD、YoloV3、Yolact、LaneNet 共四种卷积网络模型。SDK 分别对四种模型抽象出了四个类,分别是:class Ssd、class Yolo、class Yolact、class Lane。
系统环境依赖
本 SDK 需要在 64 位的 Linux Arm 系统上使用。
本 SDK API 定义在 utils/novauto/include 路径下的头文件。API 的动态链接库路径为 utils/novauto/lib64/libnovauto.so。对于第三方库的依赖,本 SDK 在 utils路径下包含了 opencv 3.4.1 版本、rknn 0.9.9 版本、yaml 0.6.3 版本,其头文件的定义分别在对应的 include 路径下,动态链接库分别在对应的 lib64 路径下。
API调用流程
1)修改配置文件,指定相应的模型参数。如,修改 SSD 模型配置文件config/SSD/config-ssd.yaml;
2) 加载模型配置文件信息。如,加载 SSD 模型LoadYamlConfig(1);
3) 创建卷积网络模型的类实例对象。如,创建 SSD 类实例对象Inference *model_instance = CreateInstance(1);
4) 读取卷积网络模型文件(后缀为.rknn)到内存,并初始化模型结构体对象。如 LoadModel(2, 1, model_info);
5) 初始化模型运行环境。如,BuildModelEnv(2, model_instance, model_info);
6) 创建推理线程。如,创建输入源为离线视频的模型推理线程CreateThread(2, video_path, "v", model_instance,model_info);
7) 释放模型运行环境。如,DestroyModelEnv(2, model_info)。
SDK下载
Nova-09 SDK和使用文档,点击下载。