search

linghu8812 / tensorrt_inference

708 stars 208 forks source link

test yolov7_trt demo failed!! #146

Open jcyhcs opened 2 years ago

jcyhcs commented 2 years ago

hi,professor: i git clone the main branch with yolov7 official, now have export.py already, so i use this script, follow your command, export onnx ok,and convert tensorrt engine with trtexec also ok, but i run demo yolo_trt, it got error : assert(engine->getNbBindings() == 2); that means the bindings more than 2, so what can i do? please help me! my tensorrt version is 7.1.3

linghu8812 commented 2 years ago

hi,professor: i git clone the main branch with yolov7 official, now have export.py already, so i use this script, follow your command, export onnx ok,and convert tensorrt engine with trtexec also ok, but i run demo yolo_trt, it got error : assert(engine->getNbBindings() == 2); that means the bindings more than 2, so what can i do? please help me! my tensorrt version is 7.1.3

use this repo https://github.com/linghu8812/yolov7, some codes has been changed. https://github.com/linghu8812/yolov7/blob/main/models/yolo.py#L62