Skip to content

RTDETRv2导出onnx时报错 #640

@Tofuinheaven

Description

@Tofuinheaven

DETR) PS D:\test\RT-DETR\RT-DETR-main> & D:\anaconda\envs\DETR\python.exe d:/test/RT-DETR/RT-DETR-main/rtdetrv2_pytorch/tools/export_onnx.py
2025-10-30 16:44:45.779961: I tensorflow/core/util/port.cc:153] oneDNN custom operations are on. You may see slightly different numerical results due to floating-point round-off errors from different computation orders. To turn them off, set the environment variable TF_ENABLE_ONEDNN_OPTS=0.
2025-10-30 16:44:47.237918: I tensorflow/core/util/port.cc:153] oneDNN custom operations are on. You may see slightly different numerical results due to floating-point round-off errors from different computation orders. To turn them off, set the environment variable TF_ENABLE_ONEDNN_OPTS=0.
Load PResNet18 state_dict
Traceback (most recent call last):
File "D:\anaconda\envs\DETR\lib\site-packages\onnx_ir\serde.py", line 99, in wrapper
return func(*args, **kwargs)
File "D:\anaconda\envs\DETR\lib\site-packages\onnx_ir\serde.py", line 1775, in serialize_tensor_into
serialize_metadata_props_into(tensor_proto.metadata_props, from.metadata_props)
AttributeError: metadata_props

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
File "d:\test\RT-DETR\RT-DETR-main\rtdetrv2_pytorch\tools\export_onnx.py", line 100, in
main(args)
File "d:\test\RT-DETR\RT-DETR-main\rtdetrv2_pytorch\tools\export_onnx.py", line 58, in main
torch.onnx.export(
File "D:\anaconda\envs\DETR\lib\site-packages\torch\onnx_init_.py", line 296, in export
return _compat.export_compat(
File "D:\anaconda\envs\DETR\lib\site-packages\torch\onnx_internal\exporter_compat.py", line 130, in export_compat
registry = registration.ONNXRegistry().from_torchlib(
File "D:\anaconda\envs\DETR\lib\site-packages\torch\onnx_internal\exporter_registration.py", line 169, in from_torchlib
torchlib_ops = onnxscript_apis.get_torchlib_ops()
File "D:\anaconda\envs\DETR\lib\site-packages\onnxscript_framework_apis\torch_2_5.py", line 86, in get_torchlib_ops
from onnxscript.function_libs.torch_lib import ( # pylint: disable=import-outside-toplevel
File "D:\anaconda\envs\DETR\lib\site-packages\onnxscript\function_libs\torch_lib\ops_init
.py", line 16, in
from . import (
File "D:\anaconda\envs\DETR\lib\site-packages\onnxscript\function_libs\torch_lib\ops\core.py", line 812, in
def _aten_as_strided_onnx(
File "D:\anaconda\envs\DETR\lib\site-packages\onnxscript\function_libs\torch_lib\registration.py", line 130, in wrapper
processed_func = onnxscript.script(opset=custom_opset)(func)
File "D:\anaconda\envs\DETR\lib\site-packages\onnxscript\main.py", line 94, in transform
result = script_check(f_ast, opset, env, src, default_opset=default_opset)
File "D:\anaconda\envs\DETR\lib\site-packages\onnxscript\main.py", line 38, in script_check
return convert.translate_function_def(f)
File "D:\anaconda\envs\DETR\lib\site-packages\onnxscript\converter.py", line 1452, in translate_function_def
fn_ir = self._translate_function_def_common(stmt)
File "D:\anaconda\envs\DETR\lib\site-packages\onnxscript\converter.py", line 1439, in _translate_function_def_common
self._translate_stmt(s, index_of_stmt=i)
File "D:\anaconda\envs\DETR\lib\site-packages\onnxscript\converter.py", line 975, in _translate_stmt
return self._translate_loop_stmt(node)
File "D:\anaconda\envs\DETR\lib\site-packages\onnxscript\converter.py", line 1247, in _translate_loop_stmt
self._translate_stmt(s)
File "D:\anaconda\envs\DETR\lib\site-packages\onnxscript\converter.py", line 961, in _translate_stmt
return self._translate_assign_stmt(node)
File "D:\anaconda\envs\DETR\lib\site-packages\onnxscript\converter.py", line 1048, in _translate_assign_stmt
assign(lhs, rhs)
File "D:\anaconda\envs\DETR\lib\site-packages\onnxscript\converter.py", line 992, in assign
t = self._translate_expr(rhs, lhs).name
File "D:\anaconda\envs\DETR\lib\site-packages\onnxscript\converter.py", line 548, in _translate_expr
r = self._translate_binary_op_expr(node)
File "D:\anaconda\envs\DETR\lib\site-packages\onnxscript\converter.py", line 859, in _translate_binary_op_expr
op, self._translate_expr(node.left), self._translate_expr(node.right)
File "D:\anaconda\envs\DETR\lib\site-packages\onnxscript\converter.py", line 558, in _translate_expr
r = self._emit_const(self._eval_constant_expr(node), target, self._source_of(node))
File "D:\anaconda\envs\DETR\lib\site-packages\onnxscript\converter.py", line 412, in emit_const
tensor = autocast.pyvalue_to_onnx_tensor(ovar, pyvalue)
File "D:\anaconda\envs\DETR\lib\site-packages\onnxscript_internal\autocast.py", line 28, in pyvalue_to_onnx_tensor
return ir.serde.serialize_tensor(ir.tensor(pyvalue, name=tensor_name))
File "D:\anaconda\envs\DETR\lib\site-packages\onnx_ir\serde.py", line 1738, in serialize_tensor
serialize_tensor_into(tensor_proto, from
=tensor)
File "D:\anaconda\envs\DETR\lib\site-packages\onnx_ir\serde.py", line 101, in wrapper
raise SerdeError(
onnx_ir.serde.SerdeError: Error calling serialize_tensor_into with: Tensor<INT64,[]>(array(1), name='int64_1')

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions