使用mmdnn將caffe模型轉換為tensorflow模型
阿新 • • 發佈:2018-11-10
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