tensorrtx加速yolov5

请教各位大佬,我把yolov5n.pt转成onnx权重文件后再转换成trt文件,在使用tensorrtx的yolov5_trt.py进行推理的时候发生了错误.

请问该怎么解决?