1. 程式人生 > >使用mmdnn將caffe模型轉換為tensorflow模型

使用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