Caffe convertir modelo mxnet --mmdnn mmdnn mencionados utilizando el modelo de conversión no fue la conversión de comandos muy claro, recientemente re-comprobado algunos, terminando en comando de conversión mmdnn.
conversión Paso
mmdnn comando de conversión paso es mmconvert, desde mmdnn / conversión / _script / convert.py, cuyos parámetros son:
comando | significado |
---|---|
-sf | Introduzca el tipo de modelo |
-en | Introduzca el modelo de la red (si el archivo independiente de la red) |
-iw | Introduzca el archivo de pesos |
-df | tipo de modelo de salida |
-si | Salida Nombre del modelo |
Durante .py generado, .npy modelo de estructura intermedia como un modelo de la forma de códigos de destino. Además, al igual que el modelo TensorFlow necesidad de especificar el nodo de salida, tales como -dstNode MMdnn_Output, pero -dstNode Yo no encontrar el valor de la clave en el convert.py no saben correr, no será un problema.
Ejemplo:
mmconvert -sf tensorflow -in imagenet_resnet_v2_152.ckpt.meta -iw imagenet_resnet_v2_152.ckpt --dstNode MMdnn_Output -df pytorch -om tf_to_pytorch_resnet_152.pth
python -m mmdnn.conversion._script.convertToIR -sf tensorflow -in imagenet_resnet_v2_152.ckpt.meta -iw imagenet_resnet_v2_152.ckpt --dstNode MMdnn_Output -df pytorch -om tf_to_pytorch_resnet_152.pth
conversión Paso
1, gire IR
Otros modelos para convertir IR mmdnn formato de comando es mmtoir, desde mmdnn / conversión / _script / convertToIR.py, qué parámetros son los siguientes:
comando | significado |
---|---|
-F | Introduzca el tipo de modelo |
-norte | Introduzca el modelo de la red (si el archivo independiente de la red) |
-w | Introduzca el archivo de pesos |
-inputShape | Tamaño de entrada (opcional) |
-nodo | Un nodo de salida (TensorFlow) |
-O | Nombre del archivo intermedio .json, .pb, .npy |
Ejemplo:
mmtoir -f tensorflow -n imagenet_resnet_v2_152.ckpt.meta -w imagenet_resnet_v2_152.ckpt --inputShape 3,112,112 -node MMdnn_Output -o converted
python -m mmdnn.conversion._script.convertToIR -f tensorflow -n imagenet_resnet_v2_152.ckpt.meta -w imagenet_resnet_v2_152.ckpt --inputShape 3,112,112 -node MMdnn_Output -o converted
2, la salida a su vez IR modelo Código
mmdnn transformar formato IR código para otra forma de la mmtocode comando modelo, desde mmdnn / conversión / _script / IRToCode.py, que parámetros son:
comando | significado |
---|---|
-F | tipo de modelo de salida |
-norte | la estructura del modelo IR |
-w | pesos modelo IR |
-O | la estructura del modelo de salida |
-dw | pesos modelo de salida (opcional) |
Ejemplo:
mmtocode -f pytorch -n converted.pb -w converted.npy -o converted_pytorch.py -dw converted_pytorch.npy
python -m mmdnn.conversion._script. IRToCode -f pytorch -n converted.pb -w converted.npy -o converted_pytorch.py -dw converted_pytorch.npy
3. modelo final de exportación
mmdnn comando Exportar el objetivo final del modelo es mmtomodel, desde mmdnn / conversión / _script / IRToModel.py, cuyos parámetros son:
comando | significado |
---|---|
-F | tipo de modelo de salida |
-en | la estructura del modelo de salida |
-iw | pesos modelo IR |
-O | Salida Nombre del modelo |
Ejemplo:
mmtomodel -f pytorch -in converted_pytorch.py -iw converted_pytorch.npy -o converted_pytorch.pth
python -m mmdnn.conversion._script. IRToModel -f pytorch -in converted_pytorch.py -iw converted_pytorch.npy -o converted_pytorch.pth