我在Jetson Nano在进行TensorRt加速过程中生成.engine文件时遇到的了如下的问题:
(yolov5) ly@ly:~/tensorrtx-master/yolov5/build$ sudo ./yolov5 -s ../yolov5s.wts yolov5s.engine s
[sudo] password for ly:
arguments not right!
./yolov5 -s // serialize model to plan file
./yolov5 -d ../samples // deserialize plan file and run inference
原因:YOLOv5权重文件和TensorRT的版本不对应导致的