search

WongKinYiu / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
GNU General Public License v3.0
13.22k stars 4.17k forks source link

onnx->tensorrt #23

Open h030162 opened 2 years ago

h030162 commented 2 years ago

when onnx_to_tensorrt In node 163 (parseGraph): INVALID_GRAPH: Assertion failed: ctx->tensors().count(inputName)

onnx is from models/export.py. my testing model is pretrained yolov7.pt

triple-Mu commented 2 years ago

You can try onn-simplifier.