[MXNet SSD model support] Problem in converting ssd inception model from mxnet to IR. · microsoft/MMdnn#62

(7 留言) (0 反應) (0 負責人)Python (5,767 star) (982 fork)batch import

enhancementhelp wanted

描述

Platform (like ubuntu 16.04):

Python version: 2.7.12

Source framework with version: MXNET 1.0.0 with GPU

Destination framework with version: Tensorflow 1.4.1 with GPU

I was able to convert InceptionV3 (mxnet) to IR but I encounter a problem when I try to convert ssd InceptionV3 model(mxnet).

Running scripts: root@2f2520263ec5:/home/bomo/Desktop# python -m mmdnn.conversion._script.convertToIR -f mxnet -n ssd_inceptionv3_512-symbol.json -d ssd_inceptionv3_512 --inputShape 3 512 512

error: [14:37:20] src/nnvm/legacy_json_util.cc:209: Loading symbol saved by previous version v0.12.0. Attempting to upgrade... [14:37:20] src/nnvm/legacy_json_util.cc:217: Symbol successfully upgraded! /mxnet/python/mxnet/module/base_module.py:53: UserWarning: You created Module with Module(..., label_names=['softmax_label']) but input with name 'softmax_label' is not found in symbol.list_arguments(). Did you mean one of: data label warnings.warn(msg) Warning: MXNet Parser has not supported operator null with name data. Warning: convert the null operator with name [data] into input layer. Warning: MXNet Parser has not supported operator null with name label. Warning: convert the null operator with name [label] into input layer. [14:37:20] /mxnet/dmlc-core/include/dmlc/./logging.h:308: **[14:37:20] src/c_api/c_api_symbolic.cc:417: InferShapeKeyword argument name data not found. Candidate arguments: [0]label** infer_shape error. Arguments: data: (1, 3, 512, 512) Traceback (most recent call last): File "/usr/lib/python2.7/runpy.py", line 174, in _run_module_as_main "__main__", fname, loader, pkg_name) File "/usr/lib/python2.7/runpy.py", line 72, in _run_code exec code in run_globals File "/usr/local/lib/python2.7/dist-packages/mmdnn/conversion/_script/convertToIR.py", line 159, in <module> _main() File "/usr/local/lib/python2.7/dist-packages/mmdnn/conversion/_script/convertToIR.py", line 154, in _main ret = _convert(args) File "/usr/local/lib/python2.7/dist-packages/mmdnn/conversion/_script/convertToIR.py", line 95, in _convert parser.gen_IR() File "/usr/local/lib/python2.7/dist-packages/mmdnn/conversion/mxnet/mxnet_parser.py", line 269, in gen_IR self.rename_UNKNOWN(current_node) File "/usr/local/lib/python2.7/dist-packages/mmdnn/conversion/mxnet/mxnet_parser.py", line 388, in rename_UNKNOWN self.set_output_shape(source_node, IR_node) File "/usr/local/lib/python2.7/dist-packages/mmdnn/conversion/mxnet/mxnet_parser.py", line 285, in set_output_shape arg_shape, output_shape, aux_shape = sym.infer_shape(data = self.data_shape) File "/mxnet/python/mxnet/symbol/symbol.py", line 965, in infer_shape res = self._infer_shape_impl(False, *args, **kwargs) File "/mxnet/python/mxnet/symbol/symbol.py", line 1095, in _infer_shape_impl ctypes.byref(complete))) File "/mxnet/python/mxnet/base.py", line 146, in check_call raise MXNetError(py_str(_LIB.MXGetLastError())) **mxnet.base.MXNetError: [14:37:20] src/c_api/c_api_symbolic.cc:417: InferShapeKeyword argument name data not found. Candidate arguments: [0]label**

貢獻者指南

技術棧: python
領域: machine learning
議題類型: bug
難度: 5
預計時間: over 1 week
活動狀態: stale
清晰度: mostly clear
前置要求: MXNet model structurePythonMMdnn conversion tool
新手友善度: 15
研究方向: Investigate the symbol file (ssd inceptionv3 512 symbol.json) to identify the actual input name, as the error suggests 'data' is not found. Compare with a working InceptionV3 model. Examine the MXNet parser in mmdnn/conversion/mxnet/mxnet parser.py, specifically the set output shape method around line 285 and the rename UNKNOWN method around line 388. The null operator warning for 'data' and 'label' indicates these nodes are being treated as inputs; ensure the parser correctly maps the model's input names. Consider updating the parser to handle dynamic input names or fallback to the first argument.