开发者手机AI - 目标识别demo

功能简介

该应用是在Openharmony 4.0系统上开发的一个目标识别的AI应用，旨在从上到下打通Openharmony AI子系统，展示Openharmony系统的AI能力，并为开发者提供AI应用的demo。

应用程序通过相机进行预览，对预览画面中的物体进行目标识别，目前该应用程序支持识别100种物体。

系统架构

下图为demo应用以及Openharmony AI子系统的架构图。

 

目标识别demo:

• JS实现UI界面的功能；
• Native接口及实现主要为JS提供接口进行AI推理。通过Native方式完成推理的前处理、推理以及后处理，这里通过调用opencv、MindSpore Lite的API接口实现主要功能；
• Mindspore Lite为Openharmony AI推理框架，为上层应用提供统一的AI推理接口，可以完成在手机等端侧设备中的模型推理过程；
• Neural Network Runtime 神经网络运行时，作为中间桥梁连通上层AI推理框架和底层加速芯片，实现AI模型的跨芯片推理计算。提供统一AI芯片驱动接口，使AI芯片驱动能够接入OpenHarmony系统。
• NNRt host 实现了NNRt HDI接口功能，通过对接底层AI芯片接口为上层应用提供NPU硬件推理的能力。

功能实现

 

• JS从相机数据流获取一张图片，调用Native的接口进行目标识别的处理。Native接口完成目标识别后将结果返回给JS，在预览中标识识别的物体。
• Native提供了4个接口：

static napi_value Init(napi_env env, napi_value exports) {
    napi_property_descriptor desc[] = {
        {"setconf", nullptr, setconf, nullptr, nullptr, nullptr, napi_default, nullptr},
        {"Init", nullptr, ObjectDectionInit, nullptr, nullptr, nullptr, napi_default, nullptr},
        {"Process", nullptr, ObjectDectionProcess, nullptr, nullptr, nullptr, napi_default, nullptr},
        {"DeInit", nullptr, ObjectDectionDeInit, nullptr, nullptr, nullptr, napi_default, nullptr}};
    napi_define_properties(env, exports, sizeof(desc) / sizeof(desc[0]), desc);
    return exports;
}复制

setconf 设置目标识别置信度的阈值，对于目标识别置信度低于阈值的结果不做显示，默认为45%。

Init、DeInit 为Native的初始化和去初始化实现，主要对模型进行创建、编译以及销毁。

Process为异步NAPI接口，主要完成目标识别的功能。实现包括前处理、推理和后处理3个部分。前处理主要完成图片预处理，使图片数据符合模型的输入格式。推理就是调用MindSpore接口实现推理。后处理主要对推理结果进行处理，使结果可以呈现到界面上。

代码路径