test yolov7_trt demo failed!!

[jcyhcs]

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]

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