更新时间:2023-03-27 GMT+08:00

开发算法模型

前提条件

目前,HiLens Kit的AI芯片支持运行“.om”模型,“.om”模型可以通过TensorFlow或Caffe模型转换而来,但“.om”模型并不支持TensorFlow和Caffe全部的算子,所以在开发模型的时候开发者需要用“.om”模型支持的算子,才能把TensorFlow和Caffe模型转换成“.om”模型。“.om”模型支持的TensorFlow和Caffe算子边界请见附录Caffe算子边界Tensorflow算子边界

“.om”模型当前暂不能完全兼容TensorFlow内置的Keras API。

“.om”模型当前不支持Caffe2。

采用ModelArts开发

ModelArts是面向AI开发者的一站式开发平台,您可以使用ModelArts开发一个用于HiLens平台的算法模型。

在使用ModelArts开发或训练的模型后,其模型将存储在ModelArts的模型列表中,详细操作指导请参见《ModelArts用户指南》。由于目前华为HiLens仅支持TensorFlow和Caffe引擎开发的算法模型,因此建议在使用ModelArts开发时,注意选择引擎类型以及模型保存格式。

ModelArts自动学习功能训练生成的模型,暂时不支持用于HiLens平台。

线下开发

线下开发指您在本地使用自己熟悉的算法模型开发工具,开发算法模型。

当前仅支持TensorFlow和Caffe引擎开发的算法模型,且您开发的模型需保存为“.pb”“.caffemodel”格式。然后再使用导入(转换)模型功能将其转换为Ascend310芯片支持的“.om”格式。

后续操作

模型开发完成后,需要把模型导入至华为HiLens控制台。针对TensorFlow和Caffe引擎开发的算法模型,会通过导入(转换)模型功能将其转换为Ascend310芯片支持的“.om”格式。具体操作请参见导入(转换)模型