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!!