• M
    Maiya

    @金天 win的安装路径不固定,一个路径搞不定所有的。所以一般再安装软件时,软件都会自动加入path,当然也可以手动加入。

    发布在 社区求助区(SOS!!) 阅读更多
  • M
    Maiya

    1 可以把nvonnxpaeser.dll 拷贝到 trtexec.exe 的目录下,【缺少什么就拷贝什么】

    2 还可以把nvonnxpaeser.dll所在目录 设置到path路径中

    发布在 社区求助区(SOS!!) 阅读更多
  • M
    Maiya

    @金天 把models/networks/msra_resnet.py中234行

                ret = {}
                for head in self.heads:
                    ret[head] = self.__getattr__(head)(x)
                return [ret]
    
    

    改为

                 res = []
                 for head in self.heads:
                     res.append(self.__getattr__(head)(x))
                 return tuple(res)
    

    就可以正常导出onnx 了,

    发布在 社区求助区(SOS!!) 阅读更多
  • M
    Maiya

    首先demo_det_r101.py已经可以正常跑通,然后我在使用export_det_r101_onnx.py时,报错

    Traceback (most recent call last):
      File "E:/LearningCodes/centernet/centernet-master/export_det_r101_onnx.py", line 191, in <module>
        detector.export_onnx(data_f, 'centernet_r101_coco.onnx')
      File "E:/LearningCodes/centernet/centernet-master/export_det_r101_onnx.py", line 179, in export_onnx
        input_names=['image'],
      File "D:\Anaconda3\envs\CenterNet\lib\site-packages\torch\onnx\__init__.py", line 25, in export
        return utils.export(*args, **kwargs)
      File "D:\Anaconda3\envs\CenterNet\lib\site-packages\torch\onnx\utils.py", line 131, in export
        strip_doc_string=strip_doc_string)
      File "D:\Anaconda3\envs\CenterNet\lib\site-packages\torch\onnx\utils.py", line 363, in _export
        _retain_param_name, do_constant_folding)
      File "D:\Anaconda3\envs\CenterNet\lib\site-packages\torch\onnx\utils.py", line 266, in _model_to_graph
        graph, torch_out = _trace_and_get_graph_from_model(model, args, training)
      File "D:\Anaconda3\envs\CenterNet\lib\site-packages\torch\onnx\utils.py", line 225, in _trace_and_get_graph_from_model
        trace, torch_out = torch.jit.get_trace_graph(model, args, _force_outplace=True)
      File "D:\Anaconda3\envs\CenterNet\lib\site-packages\torch\jit\__init__.py", line 231, in get_trace_graph
        return LegacyTracedModule(f, _force_outplace, return_inputs)(*args, **kwargs)
      File "D:\Anaconda3\envs\CenterNet\lib\site-packages\torch\nn\modules\module.py", line 493, in __call__
        result = self.forward(*input, **kwargs)
      File "D:\Anaconda3\envs\CenterNet\lib\site-packages\torch\jit\__init__.py", line 295, in forward
        out_vars, _ = _flatten(out)
    RuntimeError: Only tuples, lists and Variables supported as JIT inputs, but got dict
    

    发布在 社区求助区(SOS!!) 阅读更多
  • M
    Maiya

    @jinfagang 谢谢你们的工作,等待开源

    发布在 原创分享专区 阅读更多
  • M
    Maiya

    你好 我在运行demo_onnx.py时,先遇到:

    RuntimeError: Input must be a list of dictionaries or a single numpy array for input 'image'.

    然后,修改代码

        ort_inputs = {ort_session.get_inputs()[0].name: to_numpy(image).astype(np.float32)}
    

    就遇到新问题:
    onnxruntime.capi.onnxruntime_pybind11_state.Fail: [ONNXRuntimeError] : 1 : FAIL : Non-zero status code returned while running TopK node. Name:'TopK_1120' Status Message: k argument [1000] should not be greater than specified axis dim value [883]。

    请问如何解决?

    发布在 原创分享专区 阅读更多
  • M
    Maiya

    EmbedMask ONNX 支持c++的tensorRT加速么?

    发布在 原创分享专区 阅读更多
  • M
    Maiya

    哦哦,了解了。 费用如何呢?😁

    发布在 社区求助区(SOS!!) 阅读更多

与 ManaAI社区 的连接断开,我们正在尝试重连,请耐心等待