带出onnx出错

[zaidao2023]

使用命令： python tools/yunet2onnx.py ./configs/yunet_n.py ./weights/yunet_n.pth 导出模型时出现这个错误：

File "tools/yunet2onnx.py", line 265, in <module>
   dynamic_export=args.dynamic_export)
 File "tools/yunet2onnx.py", line 96, in pytorch2onnx
   dynamic_axes=dynamic_axes)
 File "C:\Users\jiafc\AppData\Local\conda\conda\envs\yoloAir\lib\site-packages\torch\onnx\__init__.py", line 365, in export
   export_modules_as_functions,
 File "C:\Users\jiafc\AppData\Local\conda\conda\envs\yoloAir\lib\site-packages\torch\onnx\utils.py", line 178, in export
   export_modules_as_functions=export_modules_as_functions,
 File "C:\Users\jiafc\AppData\Local\conda\conda\envs\yoloAir\lib\site-packages\torch\onnx\utils.py", line 1084, in _export
   dynamic_axes=dynamic_axes,
 File "C:\Users\jiafc\AppData\Local\conda\conda\envs\yoloAir\lib\site-packages\torch\onnx\utils.py", line 727, in _model_to_graph
   graph, params, torch_out, module = _create_jit_graph(model, args)
 File "C:\Users\jiafc\AppData\Local\conda\conda\envs\yoloAir\lib\site-packages\torch\onnx\utils.py", line 602, in _create_jit_graph
   graph, torch_out = _trace_and_get_graph_from_model(model, args)
 File "C:\Users\jiafc\AppData\Local\conda\conda\envs\yoloAir\lib\site-packages\torch\onnx\utils.py", line 518, in _trace_and_get_graph_from_model
   model, args, strict=False, _force_outplace=False, _return_inputs_states=True
 File "C:\Users\jiafc\AppData\Local\conda\conda\envs\yoloAir\lib\site-packages\torch\jit\_trace.py", line 1175, in _get_trace_graph
   outs = ONNXTracedModule(f, strict, _force_outplace, return_inputs, _return_inputs_states)(*args, **kwargs)
 File "C:\Users\jiafc\AppData\Local\conda\conda\envs\yoloAir\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
   return forward_call(*input, **kwargs)
 File "C:\Users\jiafc\AppData\Local\conda\conda\envs\yoloAir\lib\site-packages\torch\jit\_trace.py", line 95, in forward
   in_vars, in_desc = _flatten(args)
RuntimeError: Only tuples, lists and Variables are supported as JIT inputs/outputs. Dictionaries and strings are also accepted, but their usage is not recommended. Here, received an input
of unsupported type: numpy.ndarray