iwatake2222 / play_with_tensorrt

Sample projects for TensorRT in C++
Apache License 2.0
191 stars 34 forks source link

Add RTStereoNet #21

Closed kaleab-k closed 2 years ago

kaleab-k commented 2 years ago

Hi @iwatake2222,

Thanks for the great work!

Do you plan to add https://github.com/PINTO0309/PINTO_model_zoo/tree/main/165_RealtimeStereo soon?

Thank you!

iwatake2222 commented 2 years ago

No, not so far.

kaleab-k commented 2 years ago

Hey, I am trying to add it by modifying the CoEx implementation. However, I face this isuee:

[08/12/2022-20:18:37] [W] [TRT] onnx2trt_utils.cpp:363: Your ONNX model has been generated with INT64 weights, while TensorRT does not natively support INT64. Attempting to cast down to INT32.
[08/12/2022-20:18:37] [W] [TRT] onnx2trt_utils.cpp:391: One or more weights outside the range of INT32 was clamped
[08/12/2022-20:18:37] [E] [TRT] ModelImporter.cpp:748: While parsing node number 125 [Concat -> "488"]:
[08/12/2022-20:18:37] [E] [TRT] ModelImporter.cpp:749: --- Begin node ---
[08/12/2022-20:18:37] [E] [TRT] ModelImporter.cpp:750: input: "482"
input: "487"
output: "488"
name: "Concat_261"
op_type: "Concat"
attribute {
  name: "axis"
  i: 4
  type: INT
}

[08/12/2022-20:18:37] [E] [TRT] ModelImporter.cpp:751: --- End node ---
[08/12/2022-20:18:37] [E] [TRT] ModelImporter.cpp:753: ERROR: onnx2trt_utils.cpp:334 In function convertAxis:
[8] Assertion failed: (axis >= 0 && axis <= nbDims) && "Axis must be in the range [0, nbDims]."
[ERR: InferenceHelperTensorRt][149] Failed to parse onnx file (../play_with_tensorrt/pj_tensorrt_depth_stereo_rtstereo/build/resource//model/rtstereonet_maxdisp192_480x640.onnx)[ERR: DepthStereoEngine][91] Inference helper is not created
Initialization Error

Do you have any idea how I can solve this issue?

iwatake2222 commented 2 years ago

You already got a solution https://github.com/PINTO0309/PINTO_model_zoo/issues/278