yolov5转模型成功的
第一步下载并把模型文件复制到这个项目下:
https://github.com/linghu8812/yolov5(用这个代码可以转成功)
第二步:(或者不要第一步也行,直接在官方的代码执行以下操作)
python3 export.py —weights weights/yolov5s.pt --imgsz 640 --include onnx --simplify
第三步:
trtexec --onnx=yolov5s.onnx --saveEngine=yolov5s.engine
pt->onnx->engine
于 2023-01-20 14:38:41 首次发布