onnx 转 mxnet 保存静态文件

俭任G

已于 2022-01-20 09:33:08 修改

阅读量520

点赞数

分类专栏： MXNet 文章标签： mxnet 深度学习

于 2022-01-19 10:09:34 首次发布

本文链接：https://blog.csdn.net/weixin_41798111/article/details/122574590

版权

MXNet 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

将onnx转为mxnet可读取的静态模型及参数

sym, arg, aux = onnx_mxnet.import_model(onnx_file)
model = mx.mod.Module(sym,data_names=('input',),label_names=('output',))
sym.save(path_json)
data = np.zeros(shape=(B,C,D,H,W))
trainiter = mx.io.NDArrayIter(data=data,batch_size=batch_size,data_name='input')
model.bind(data_shapes=trainiter.provide_data,for_training=False,force_rebind=False)
model.init_params(arg,aux,allow_missing=True,allow_extra=True)
model.save_params(params_path)