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

LAVAL小助手
发布于 2024-4-11 15:16
浏览
0收藏

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

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

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

开发者手机AI - 目标识别demo-鸿蒙开发者社区

目标识别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硬件推理的能力。

功能实现

开发者手机AI - 目标识别demo-鸿蒙开发者社区

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接口实现推理。后处理主要对推理结果进行处理,使结果可以呈现到界面上。

代码路径
开发者手机AI - 目标识别demo-鸿蒙开发者社区

收藏
回复
举报
回复
    相关推荐