⼈⼯智能⼯具链转换平台

我要开发同款
allan08082021年12月28日
71阅读

作品详情

模型转换编译⼯具是使⽤python, C++开发的程序。作⽤是把ONNX, Tensflow, Caffe平台训练好的模型,转换成NPU模型。这个模型是npu应⽤程序唯⼀识别的⽹络结构模型,是应⽤SDK运⾏的必要参数。模型转换编译⼯具主要包含两个部分:数据定点化和⽹络结构的转换。常⻅的深度学习开发⼯具如Pytorch, Tesorflow, Caffe模型中包含每⼀个layer/ops训练好的参数,但是这些参数都是浮点型的,⽽NPU只⽀持定点数据,所以,我们需要根据相关配置⽂件,对这些数据进⾏定点化。数据结构的转换,主要是把Pytroch, Tensorflow, Caffe模型中每⼀个layer/ops的相关配置信息,转换成NPU⾃定义的⼀套数据结构,同时,根据需要增加⼀些额外的信息,供NPU使⽤。最后将定点信息和转换后的⽹络结构,按照⼀定的格式保存到⽂件中,即npumodel⽂件。
查看全文
声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论