MXNet tools in docker

How to convert MXNet model to Apple CoreML:

docker run -v "$PWD":/data --rm -it dharmeshkakadia/mxnet-coreml-tools-docker python mxnet_coreml_converter.py 

For example, if you want to convert Squeezenet model to coreml, to use with iOS. Run the following from the a directory containing Squeezenet model files (Params, symbols, labels) and will generate squeezenetv11.mlmodel in the current directory.

docker run -v "$PWD":/data --rm -it dharmeshkakadia/mxnet-coreml-tools-docker python mxnet_coreml_converter.py --model-prefix='/data/squeezenet_v1.1' --epoch=0 --input-shape='{"data":"3,227,227"}' --mode=classifier --pre-processing-arguments='{"image_input_names":"data"}' --class-labels /data/synset.txt --output-file="/data/squeezenetv11.mlmodel"

The Dockerfile is located on github.