windows下yolov8训练环境配置
安装
anaconda3
conda create -n yolov8 python=3.8
## 进入yolov8 虚拟环境
activate yolov8
pip install ultralytics
conda install chardet
安装pytorch、cuda
https://pytorch.org/get-started/locally/
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
安装onnx格式支持
conda install onnx
导出pt模型为onnx格式:
yolo export model=best.pt format=onnx
https://docs.ultralytics.com/tasks/detect/#models
训练:
yolo detect train data=my_coco.yaml model=yolov8s.pt epochs=150 imgsz=640 batch=64 workers=0 device=0
本文作者:一颗西瓜糖
本文链接:https://www.cnblogs.com/ykxgt/p/18361059
版权声明:本作品采用知识共享署名-非商业性使用-禁止演绎 2.5 中国大陆许可协议进行许可。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步