流程说明:
- 建模
根据目标任务,如检测、分类、分割等,收集相关数据标注、清洗 、预处理。选择合适的深度学习框架,如tensorflow/pytorch等,对数据进行建模,最终得到pb/pt等各种格式的模型。 - 模型转换
海思目前仅支持caffe->nnie的模型转换,所以需要先将我们自己训练的模型转换为caffe格式,即pb/pt->onnx(可选)->caffe->nnie。过程中onnx转caffe需要pycaffe环境,如跳过onnx,可忽略。onnx转nnie需要使用华为官方工具Ruyistudio(支持nnie1.1/1.2)/Aistudio(支持nnie1.3)或者华为好望商城线上转换或者linux转换环境(需要svp开发包)。 - 模型加载推理
调用华为文件服务接口加载推理nnie模型,可通过vscode/mobaxterm连接后台调试算法。