apache / mxnet

Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more
https://mxnet.apache.org
Apache License 2.0
20.77k stars 6.8k forks source link

[Feature Request] Add copy operator to MXNet to CoreML convertor #14003

Open ChaiBapchya opened 5 years ago

ChaiBapchya commented 5 years ago

As revealed in the issue #13965 currently only 12 operators are supported to convert from MXNet to CoreML.

    'FullyConnected' : _layers.convert_dense,
    'Activation'     : _layers.convert_activation,
    'SoftmaxOutput'  : _layers.convert_softmax,
    'Convolution'    : _layers.convert_convolution,
    'Pooling'        : _layers.convert_pooling,
    'Flatten'        : _layers.convert_flatten,
    'transpose'      : _layers.convert_transpose,
    'Concat'         : _layers.convert_concat,
    'BatchNorm'      : _layers.convert_batchnorm,
    'elemwise_add'   : _layers.convert_elementwise_add,
    'Reshape'        : _layers.convert_reshape,
    'Deconvolution'  : _layers.convert_deconvolution,

Based on the user request, we could add copy and similar commonly needed ops as well in this file

incubator-mxnet/tools/coreml/converter/_layers.py

@mxnet-label-bot add [Feature request]

mxnet-label-bot commented 5 years ago

Hey, this is the MXNet Label Bot. Thank you for submitting the issue! I will try and suggest some labels so that the appropriate MXNet community members can help resolve it. Here are my recommended labels: Feature

frankfliu commented 5 years ago

@mxnet-label-bot add [feature request]

markproy commented 5 years ago

Not sure how I add "+1" to this request, so I'm just adding this comment. I'd like to convert a ResNet-50 model created by MXNet under the hood of SageMaker's Image Classification built-in algo. Unfortunately, it uses "_copy" (and perhaps other unsupported layers). Thus, the converter is unable to create a coreML model.

tapanh commented 5 years ago

I have the exact same use case as @markproy above. Would be great to have this feature!

mn033 commented 5 years ago

@markproy I have exactly the same issue with SageMaker Object Detection. Did you solve this problem in any way? (other converter etc.) ?

tycoles commented 4 years ago

Same problem here!