Closed cyrusbehr closed 2 years ago
When I use version TensorRT-7.1.3.4 of tensorrt, I get the following crash message:
(base) cyrus:~/work/repos/tensorrt_inference/RetinaFace/build[master !] ./RetinaFace_trt ../config.yaml ../samples
----------------------------------------------------------------
Input filename: ../R50.onnx
ONNX IR version: 0.0.8
Opset version: 15
Producer name:
Producer version:
Domain:
Model version: 0
Doc string:
----------------------------------------------------------------
[11/17/2021-18:10:53] [W] [TRT] onnx2trt_utils.cpp:220: Your ONNX model has been generated with INT64 weights, while TensorRT does not natively support INT64. Attempting to cast down to INT32.
ERROR: builtin_op_importers.cpp:2523 In function importResize:
[8] Assertion failed: (mode != "nearest" || nearest_mode == "floor") && "This version of TensorRT only supports floor nearest_mode!"
[11/17/2021-18:10:53] [E] Failure while parsing ONNX file
start building engine
[11/17/2021-18:10:53] [E] [TRT] Network must have at least one output
[11/17/2021-18:10:53] [E] [TRT] Network validation failed.
build engine done
RetinaFace_trt: /home/cyrus/work/repos/tensorrt_inference/RetinaFace/../includes/common/common.hpp:138: void onnxToTRTModel(const string&, const string&, nvinfer1::ICudaEngine*&, const int&): Assertion `engine' failed.
Aborted (core dumped)
That resolved the issue. Thank you.
Hi~ How did you end up solving this problem?
@buaaxiejun the issue is resolved by ensuring you have onnx version 1.5.0 installed before running the export_onnx.py
script.
Followed instructions for converting mxnet retinaface model to onnx format. I built the
RetinaFace_trt
executable, linking against TensorRT-8.0.3.4.When I try to run the exe, I get the following crash: