search

linghu8812 / tensorrt_inference

705 stars 207 forks source link

`engine->getNbBindings() == 2' failed #3

Closed ZhangChaoZhong closed 3 years ago

ZhangChaoZhong commented 3 years ago

Hello! I have such a problem: as shown in the figure below. Can you help me solve it? thank you! ![图片](https://user-images.githubusercontent.com/30583522/99945482-aa065300-2daf-11eb-979c-3ef186561136.png

linghu8812 commented 3 years ago

@ZhangChaoZhong Hello，please export yolov5 onnx with the instruction in the yolov5/README, use my yolov5 repo to export yolov5 onnx instead of the official yolov5. The yolo.py file has a little difference

ZhangChaoZhong commented 3 years ago

Thank you for your answer.