ML — 机器学习相关的操作

class DataInput(...)

使用示例:

import haascv
from haascv import ml
print("-------------------Welcome HaasAI MicroPython--------------------");
print("-----ml odla demo start-----");
mlobj = ml();
mlobj.open(ml.ML_ENGINE_ODLA);
mlobj.setInputData("/data/odla/test4.bmp");
mlobj.loadNet("default");
mlobj.predict();
responses_value = bytearray(10);
mlobj.getPredictResponses(responses_value);
print(responses_value);
mlobj.unLoadNet();
mlobj.close();
print("-----ml odla demo end-----");

常量

当前支持的推理引擎的种类:
ML_ENGINE_NONE
ML_ENGINE_MNN
ML_ENGINE_CLOUD
ML_ENGINE_ODLA
ML_ENGINE_MAX

操作函数

open(MLEngineType_t)

打开推理引擎功能模块

close()

关闭推理引擎功能模块

config(key,secret,region_id,endpoint,url)

配置推理引擎需要的参数

setInputData(mFileName)

设置输入数据

loadNet(mFileName)

加载AI 模型文件

predict()

开始推理

getPredictResponses()

获取推理结果

unLoadNet()

卸载AI模型文件