[GitHub] pracheer opened a new pull request #7337: MXNet -> Apple CoreML converter (Review-only. Don't merge)

[gi...@git.apache.org]

pracheer opened a new pull request #7337: MXNet -> Apple CoreML converter (Review-only. Don't merge)
URL: https://github.com/apache/incubator-mxnet/pull/7337
 
 
   This tool helps convert MXNet models into [Apple CoreML](https://developer.apple.com/documentation/coreml) format which can then be run on Apple devices.
   
   This PR is currently for review only since we are in the process of figuring out the licensing issues with Apple. We got the initial code from them and they have handed over the maintenance and development of this tool to MXNet community from now on.
   
   This has been tested against MXNet 0.10.0 as well as against the current master branch.
   
   Example of a sample command to convert squeeze-net model.?
   `python mxnet_coreml_converter.py --model-prefix='squeezenet_v1.1' --epoch=0 --input-shape='{"data":"3,224,224"}' --output-file="squeezenet_v11.mlmodel"`
   
   This tools is currently able to convert Inception/Resnet/VGG/Squeezenet models. Support for more vision models as well as nlp models will be added later.
   
   Unit tests have been added which test 1/ each layer individually 2/ some of the standard models from mxnet zoo against random data as well as on a subset of imagenet data. In all the tests we compare the accuracy of MXNet vs that of CoreML using the converted model.
   
   @mli @piiswrong @madjam @arank @jiajiechen @indhub
 
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services