pt->onnx->engine

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

你可能感兴趣的:(人工智能,python,深度学习)