我在尝试运行基于我从 onnx 文件生成的 yolov4.engine 的代码时遇到问题,我收到此错误:
[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
我在 jetson AGX Xavier Dev Kit 上运行它,代码在我的主机上运行良好,我只有在我的 jetson 卡上运行它时才得到错误。
在我的主机上我有:Tensorrt 7.2.3-1+cuda11.1
在我的 jetson AGX Xavier 上:Tensorrt 7.1.3-1+cuda10.2 它与最后一个带有 sdk 管理器的 jetpack 4.5.1 一起安装
我什至尝试用 qt 创建一个新项目,并编写了相同的代码,我添加了 yolov4 需要运行的文件,它编译得很好,但是当我执行代码时出现错误。
有谁知道我真的被卡住了
谢谢。