使用mmdnn将caffe模型转换为tensorflow模型

mmdnn是微软推出的用于各个模型互转的工具,支持主流框架,同类产品有onnx.

这两个存在的共同问题是文档太旧,更新后很多命令不再适用

这里给出mmdnn能用的命令

@echo off
set PYTHONPATH=D:/CNN/caffe/python
mmconvert -sf caffe -in squeezenet-deploy.prototxt -w squeezenet.caffemodel -df tensorflow -om squeezenet --inputShape 3,227,227
rem python -m mmdnn.conversion._script.convertToIR -f caffe -d kit_imagenet -n squeezenet-deploy.prototxt -w squeezenet.caffemodel --inputShape 3,227,227
rem python -m mmdnn.conversion._script.IRToCode -f tensorflow --IRModelPath kit_imagenet.pb --dstModelPath kit_imagenet.py -w kit_imagenet.npy
rem python -m mmdnn.conversion.examples.tensorflow.imagenet_test -n kit_imagenet.py -w kit_imagenet.npy --dump ./caffe_alexnet.ckpt --dump_tag SERVING
pause

猜你喜欢

转载自blog.csdn.net/minstyrain/article/details/83149124
今日推荐