YOLOv7_Tensorrt
YOLOv7_Tensorrt copied to clipboard
A simple implementation of Tensorrt YOLOv7
已生成出yolov7_fp16.engine,但出現如下錯誤 Traceback (most recent call last): File "/jupyter/local/Github/sreaming/tensorrt/infer.py", line 113, in trt_engine = TRT_engine("./yolov7_fp16.engine") File "/jupyter/local/Github/sreaming/tensorrt/infer.py", line 13, in __init__ self.init_engine() File "/jupyter/local/Github/sreaming/tensorrt/infer.py", line 25, in init_engine for index in...
博主您好,看了您的教程在用tensorrt生成engine模型时正常,但是在infer时会报错,想问下是什么原因? 另外我在用官方的export时导出onnx模型再生成engine时不会报这个错,但是infer出来的图片预测框全堆在一起,想问下是什么原因。  