From ac9db5b4e7ae5dd2c77481306577e066f34be22f Mon Sep 17 00:00:00 2001 From: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com> Date: Wed, 31 Aug 2022 15:48:08 +0800 Subject: [PATCH] [Fix] Close onnx optimizer for ncnn (#961) * close onnx optimizer for ncnn * fix docformatter * fix lint --- mmdeploy/apis/pytorch2onnx.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/mmdeploy/apis/pytorch2onnx.py b/mmdeploy/apis/pytorch2onnx.py index e7bef9d23..832a85b5d 100644 --- a/mmdeploy/apis/pytorch2onnx.py +++ b/mmdeploy/apis/pytorch2onnx.py @@ -6,8 +6,8 @@ import mmcv import torch from mmdeploy.apis.core.pipeline_manager import no_mp -from mmdeploy.utils import (get_backend, get_dynamic_axes, get_input_shape, - get_onnx_config, load_config) +from mmdeploy.utils import (Backend, get_backend, get_dynamic_axes, + get_input_shape, get_onnx_config, load_config) from .core import PIPELINE_MANAGER from .onnx import export @@ -88,6 +88,10 @@ def torch2onnx(img: Any, keep_initializers_as_inputs = onnx_cfg.get('keep_initializers_as_inputs', True) optimize = onnx_cfg.get('optimize', False) + if backend == Backend.NCNN.value: + """NCNN backend needs a precise blob counts, while using onnx optimizer + will merge duplicate initilizers without reference count.""" + optimize = False with no_mp(): export( torch_model,