Skip to main content

Tool to convert MXNet models into Apple CoreML model format.

Project description

This tool helps convert MXNet models into Apple CoreML format which can then be run on Apple devices. You can find more information about tool on our github page.

Prerequisites

This package can only be installed on MacOS X since it relies on Apple’s CoreML SDK. This tool can be run on MacOS 10.12 or higher though for running inferences on the converted model MacOS 10.13 or higher is needed (or for phones, iOS 11 or above).

Installation

The method for installing this tool follows the standard python package installation steps. Once you have set up a python environment, run:

pip install mxnet-coreml-converter

The package documentation contains more details on how to use coremltools.

Dependencies

This tool has the following dependencies: * pyyaml (3.12+) * mxnet (0.10.0+) * coremltools (0.5.1+)

Sample Usage

In order to convert, say a Squeezenet model, with labels from synset.txt, execute this

mxnet_coreml_converter.py --model-prefix='squeezenet_v1.1' \
--epoch=0 --input-shape='{"data":"3,227,227"}' \
--mode=classifier --pre-processing-arguments='{"image_input_names":"data"}' \
--class-labels synset.txt --output-file="squeezenetv11.mlmodel"

More Information

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

mxnet_coreml_converter-0.1.0a7-cp27-cp27m-macosx_10_7_x86_64.whl (17.3 kB view hashes)

Uploaded CPython 2.7m macOS 10.7+ x86-64

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page