mmdnn实现模型转换

最新推荐文章于 2024-08-07 10:08:40 发布

dbmeng

最新推荐文章于 2024-08-07 10:08:40 发布

阅读量1.3k

点赞数

分类专栏： general

本文链接：https://blog.csdn.net/dbmeng/article/details/90759609

版权

本文档介绍了如何使用mmdnn将模型从mxnet转换为IR格式，再由IR转换为pytorch模型。首先，通过mmtoir将mxnet模型转为IR，接着用mmtocode将IR转换为pytorch代码和权重文件，最后使用mmtomodel导出pytorch模型。

摘要由CSDN通过智能技术生成

1、转IR

mmtoir/python3 -m mmdnn.conversion._script.convertToIR

python3 -m mmdnn.conversion._script.convertToIR -f mxnet -n model-symbol.json -w model-0000.params --inputShape 3,112,112 -o IR34

输出：

UserWarning: You created Module with Module(..., label_names=['softmax_label']) but input with name 'softmax_label' is not found in symbol.list_arguments(). Did you mean one of:
data
warnings.warn(msg)
Warning: MXNet Parser has not supported operator null with name data.
Warning: convert the null operator with name [data] into input layer.
IR network structure is saved as [IR34.json].
IR network structure is saved as [IR34.pb].
IR weights are saved as [IR34.npy]

2、IR转输出模型Code

mmtocode/python3 -m mmdnn.conversion._script.IRToCode

python3 -m mmdnn.conversion._script.IRToCode -f pytorch --IRModelPath IR34.pb --IRWeightPath IR34.npy -o IR34-2.py -dw IR34-2.npy