yolov导出engin推理加速
·
最新版yolov5 支持直接导出engin
parser.add_argument('--include',
nargs='+',
default=['torchscript'],
help='torchscript, onnx, openvino, engine, coreml, saved_model, pb, tflite, edgetpu, tfjs')
在anaconda窗口切换换对应环境(我的环境名称是torch_test)
activate torch_test
cd 到yolov5文件夹里面 (E:\Pycharm\py\yolov5 是我的文件夹地址 修改成自己的yolov5地址)
cd E:\Pycharm\py\yolov5
如果不是c盘 还需要切换盘
d:
然后来到yolov5文件里面输入命令
python export.py --device 0 --weights yolov5s.pt --include engine --half
--weights 参数是你的pt文件
--include 参数是选择是engine格式 还支持
'torchscript, onnx, openvino, engine, coreml, saved_model, pb, tflite, edgetpu, tfjs'

导出完毕
yolov目标检测交流群:732818397
更多推荐

所有评论(0)