Skip to content
This repository was archived by the owner on Mar 26, 2019. It is now read-only.
This repository was archived by the owner on Mar 26, 2019. It is now read-only.

ONNX model zoo model fails to be imported #48

@lupesko

Description

@lupesko

Importing the model below to MXNet errors out.

Model: FER+ Emotion Recognition
To reproduce:

  • Install MXNet Model Server pip install mxnet-model-server
  • Download the ONNX model listed above into the working directory, add signature.json and synset.txt
  • Export the ONNX model into MMS model archive mxnet-model-export --model-name emotion-detection --model-path .

Expected result: emotion-detection.model file is created

Actual result: error in the onnx-mxnet code:

Traceback (most recent call last):
  File "/Users/hag/anaconda/envs/mms-onnx-talk-3-6/bin/mxnet-model-export", line 11, in <module>
    sys.exit(export())
  File "/Users/hag/anaconda/envs/mms-onnx-talk-3-6/lib/python3.6/site-packages/mms/export_model.py", line 377, in export
    export_model(args.model_name, args.model_path, args.service_file_path)
  File "/Users/hag/anaconda/envs/mms-onnx-talk-3-6/lib/python3.6/site-packages/mms/export_model.py", line 352, in export_model
    symbol_file, params_file = convert_onnx_model(model_path, onnx_file)
  File "/Users/hag/anaconda/envs/mms-onnx-talk-3-6/lib/python3.6/site-packages/mms/export_model.py", line 237, in convert_onnx_model
    sym, params = onnx_mxnet.import_model(os.path.join(model_path, onnx_file))
  File "/Users/hag/anaconda/envs/mms-onnx-talk-3-6/lib/python3.6/site-packages/onnx_mxnet/__init__.py", line 35, in import_model
    sym, params = graph.from_onnx(model_proto.graph)
  File "/Users/hag/anaconda/envs/mms-onnx-talk-3-6/lib/python3.6/site-packages/onnx_mxnet/import_onnx.py", line 120, in from_onnx
    mx_attr = self._fix_channels(new_op, mx_attr, list(node.input))
  File "/Users/hag/anaconda/envs/mms-onnx-talk-3-6/lib/python3.6/site-packages/onnx_mxnet/import_onnx.py", line 308, in _fix_channels
    weight_name = self._renames[inputs[1]]
KeyError: 'Parameter3'

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions