环境
pip uninstall ultralytics
pip install --upgrade ultralytics
1,训练
yolo train data=coco128.yaml model=yolov8s.pt epochs=10000 lr0=0.01
转化 onnx
yolo mode=export model=C:\Users\Administrator\Desktop\yt98.pt format=onnx dynamic=True simplify=True
转化 trt
trtexec.exe --onnx=yt98.onnx --saveEngine=yt98.trt --buildOnly --minShapes=images:1x3x640x640 --optShapes=images:4x3x640x640 --maxShapes=images:8x3x640x640
v8_更新后的训练方法,以及转trt方法
最新推荐文章于 2024-06-19 15:30:18 发布