最新版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

更多推荐