mxnet convert to tensorflow

mxnet convert to tensorflow


参考链接: [ https://github.com/microsoft/MMdnn]

第一步:下载最新的mmdnn版本

pip install -U git+https://github.com/Microsoft/MMdnn.git@master

第二步:将模型转换为过渡的表达形式,生成npy文件
需要json和params文件

python -m mmdnn.conversion._script.convertToIR -f mxnet -n model-symbol.json -w model-0000.params -d resnet50 --inputShape 3,112,112

第三步:将过渡形式转换为tensorflow格式,同时将模型参数和结构保存到pb文件中

python -m mmdnn.conversion._script.IRToCode -f tensorflow --IRModelPath resnet50.pb --IRWeightPath resnet50.npy --dstModelPath tf_resnet50.py

可能会遇到的问题:

File "/usr/lib/python3.6/site-packages/mmdnn/conversion/mxnet/mxnet_parser.py", line 398, in rename_FullyConnected
    weight = self.weight_data.get(source_node.name + "_weight").asnumpy().transpose((1, 0))
AttributeError: 'NoneType' object has no attribute 'asnumpy'

原因是source_node.name和原mxnet中最后一层变量名称不一致,需要手动更改mxnet_parser.py。

你可能感兴趣的:(mxnet convert to tensorflow)