yaoyi30 / yolov5_tensorrt_c- Goto Github PK

望及时修复

使用yolov5-6.1自训练了一个模型，转成onnx后去推理，重新编译运行后没有报错，但是结果图片上没有画框，

您好,我想請問一下如何解決這個問題

[06/09/2023-10:44:46] [W] [TRT] onnx2trt_utils.cpp:198: Your ONNX model has been generated with INT64 weights, while TensorRT does not natively support INT64. Attempting to cast down to INT32.
[06/09/2023-10:44:46] [W] [TRT] onnx2trt_utils.cpp:198: Your ONNX model has been generated with INT64 weights, while TensorRT does not natively support INT64. Attempting to cast down to INT32.
[06/09/2023-10:44:46] [W] [TRT] onnx2trt_utils.cpp:198: Your ONNX model has been generated with INT64 weights, while TensorRT does not natively support INT64. Attempting to cast down to INT32.
[06/09/2023-10:44:46] [W] [TRT] onnx2trt_utils.cpp:198: Your ONNX model has been generated with INT64 weights, while TensorRT does not natively support INT64. Attempting to cast down to INT32.
[06/09/2023-10:44:46] [W] [TRT] onnx2trt_utils.cpp:222: One or more weights outside the range of INT32 was clamped
[06/09/2023-10:44:46] [W] [TRT] onnx2trt_utils.cpp:198: Your ONNX model has been generated with INT64 weights, while TensorRT does not natively support INT64. Attempting to cast down to INT32.
ERROR: ModelImporter.cpp:92 In function parseGraph:
[8] Assertion failed: convertOnnxWeights(initializer, &weights, ctx)
[06/09/2023-10:44:46] [E] Failure while parsing ONNX file
start building engine
[06/09/2023-10:44:46] [E] [TRT] Network must have at least one output
[06/09/2023-10:44:46] [E] [TRT] Network validation failed.
build engine done
yolov5_trt: /home/lewin/yolov5_TensorRT_C-/src/common/common.hpp:138: void onnxToTRTModel(const string&, const string&, nvinfer1::ICudaEngine*&, const int&): Assertion `engine' failed.
已經終止 (核心已傾印)