1.模型转换

keras框架的.h5文件转换为onnx格式的模型文件

1、.h5格式转换为onnx格式,借助onnxmltools以及MagicONNX
2、onnx格式按照act 终端命令行转换为.om格式的文件,需要指定NCHW格式的输入数据,以及input_name:1,1,28,28
3、inputname需要根据nutron可视化onnx模型,以便获得onnx模型的输入层的名字,即name
此部分的依赖包:onnxmltools,onnxconverter_common,tf2onnx,numpy<1.24.4版本,用np1.19.5即可
MagicONNX 要从git上开源工作项目中获得。下载及安装流程为:https://github.com/jadehh/MagicONNX/tree/dev

come on!!

Logo

昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链

更多推荐