MXNet to ONNX export bug

[ehsanmok]

When trying to convert yolov3_mobilenetv1.0_coco pretrained from gluonCV v0.5 to ONNX via onnx_mxnet.export_model (using mxnet-cu90mkl==1.5.0b20190313 ), I get the following error:

/anaconda3/lib/python3.7/site-packages/mxnet/contrib/onnx/mx2onnx/_op_translations.py in convert_slice_axis(node, **kwargs)
   1320     axes = int(attrs.get("axis"))
   1321     starts = int(attrs.get("begin"))
-> 1322     ends = int(attrs.get("end", None))
   1323     if not ends:
   1324         raise ValueError("Slice: ONNX doesnt't support 'None' in 'end' attribute")

ValueError: invalid literal for int() with base 10: 'None'

This matches this line which is a bug introduced in #12878 .

[mxnet-label-bot]

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: ONNX, Bug

[lanking520]

@Roshrini Could you please take a look?

[AnaRhisT94]

@ehsanmok mxnet isn't responsible for this error, it's only ONNX.
update your ONNX version, should work when ONNX==1.2.2

[ehsanmok]

@AnaRhisT94 No, my ONNX is already the latest v1.5.0. It's when calling export_model causes that to happen. int(None) is never valid.

[AnaRhisT94]

Try to use ONNX 1.2.2

[ehsanmok]

Same error with ONNX 1.2.2

[vandanavk]

@mxnet-label-bot add [Bug]

@ehsanmok I'm looking into this, could you share your script?

[AnaRhisT94]

Same error with ONNX 1.2.2

I see, well just delete the None then?

[ehsanmok]

@vandanavk here is an MVE

from os import path as osp
import numpy as np
import mxnet as mx
from mxnet.contrib import onnx as onnx_mxnet
from mxnet import gluon
from gluoncv import model_zoo, data, utils

OUTPUT = "./output"
DATA = "./data/cat.png"
SIZE = 320
MODEL = "yolo3_mobilenet1.0_coco"
INPUT_SHAPE = (1, 3, SIZE, SIZE)

net = model_zoo.get_model(MODEL, pretrained=True)
net.hybridize()
# pass an img to trigger init after hybridize
x, _ = data.transforms.presets.yolo.load_test(DATA, short=SIZE)
_, _ = net(x)

net.export(osp.join(OUTPUT, MODEL))
sym = osp.join(OUTPUT, MODEL + "-symbol.json")
params = osp.join(OUTPUT, MODEL + "-0000.params")
onnx_file = osp.join(OUTPUT, MODEL + ".onnx")

converted_model_path = onnx_mxnet.export_model(sym, params, [INPUT_SHAPE], np.float32, onnx_file, verbose=True)

[vandanavk]

@ehsanmok I tried the following code with the PR #14942. ValueError: invalid literal for int() with base 10: 'None' error doesn't occur anymore but I do see AttributeError: No conversion function registered for op type _arange yet.. _arange export can be filed as separate feature request. Please try PR #14942 and let me know if it works for you.

from os import path as osp
import numpy as np
import mxnet as mx
from mxnet.contrib import onnx as onnx_mxnet
from mxnet import gluon
from gluoncv import model_zoo, data, utils

OUTPUT = "./"
DATA = "./cat.jpg"
SIZE = 320
MODEL = "yolo3_darknet53_coco"
INPUT_SHAPE = (1, 3, SIZE, SIZE)

net = model_zoo.get_model(MODEL, pretrained=True)
net.hybridize()
# pass an img to trigger init after hybridize
x, _ = data.transforms.presets.yolo.load_test(DATA, short=SIZE)
_ = net(x)

net.export(osp.join(OUTPUT, MODEL))
sym = osp.join(OUTPUT, MODEL + "-symbol.json")
params = osp.join(OUTPUT, MODEL + "-0000.params")
onnx_file = osp.join(OUTPUT, MODEL + ".onnx")

converted_model_path = onnx_mxnet.export_model(sym, params, [INPUT_SHAPE], np.float32, onnx_file, verbose=True)

[bloatybo]

@ehsanmok I met the same problem with you. I saw the issue has been around for 3 months. Did you solve it?

[vandanavk]

@ehsanmok I met the same problem with you. I saw the issue has been around for 3 months. Did you solve it?

Can you try the PR #14942? I dint see the issue with this PR.