YOLOv4 on tensorRT INVALID_ARGUMENT: Cannot find binding of given name: nmsed_classes

I have an issue while trying to run my code based on yolov4.engine that i generated from my onnx file, and i get this error:

[E] [TRT] INVALID_ARGUMENT: Cannot find binding of given name: num_detections

[E] [TRT] INVALID_ARGUMENT: Cannot find binding of given name: nmsed_boxes

[E] [TRT] INVALID_ARGUMENT: Cannot find binding of given name: nmsed_scores

[E] [TRT] INVALID_ARGUMENT: Cannot find binding of given name: nmsed_classes

I am running this on jetson AGX Xavier Dev Kit and the code works fine on my host machine, i got the error only at the moment when i run it on my jetson card.

on my host machine i have :
Tensorrt 7.2.3-1+cuda11.1

on my jetson AGX Xavier:
Tensorrt 7.1.3-1+cuda10.2 it was installed with last jetpack 4.5.1 with sdk manager

i even tried to create a new project with qt and i wrote the same code, i added the files yolov4 needs to run, and it compiles fine but i have the error when i execute the code.

Does anyone has an idea i am really stuck

Thanks.

Source: Windows Questions C++

LEAVE A COMMENT