第一:安装miniconda
网址:Index of /anaconda/miniconda/ 登录网址后
在网页按ctr+F 输入:搜py38 Miniconda3-py38_22.11.1-1-Windows-x86_64.exe 52.5 MiB 2022-12-23 07:57 下载进行安装 安装过程中记得加环境变量这个项。
第二:创建环境
命令:conda create -n yolov8 python=3.8
激活:命令:conda activate yolov8
第三:配置清华源(想下载快点)清华源:https:/mirrors.tuna.tsinghua.edu.cn/help/pypi/
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
第四:安装pytorch
https://pytorch.org/
如果只需要训练,简单推理,则无需要单独安装CUDA,直接安装pytorch;如果有部署需求,例如导出TensorRT模型,则需要进行CUDA安装
# CUDA 11.6
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia
# CUDA 11.7
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia
# CPU Only
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 cpuonly -c pytorch
清除缓存:运行 conda clean --all 清除 Conda 的缓存,然后重试安装
第五\测试运行:yolo predict model=yolov8n.pt source=ultralytics/assets/bus.jpg
第六\删除ultralytics pip uninstall ultralytics
第七:\源码安装: 在当前文件目录中 pip install -e. 查看安装情况:pip list
训练
1、conda activate yolov8
2、pip install labelimg
3、打开labelimg
4、打开图片open dir
5、安装完成注意的问题:打开view ->Auto save mocie
YOLO SAVE
6、标注,右键Create RectBox
整理数据集 首选数据集放在源码:ultralytics-main----》 datasetsg下面label s文件夹也是
打开: labelimg
训练:yolo task=detect mode=train model=./yolov8n.pt data=yolo-bvn.yaml epochs=30 workers=1 batch=16
预测:yolo detect predict model=runs/detect/train/weights/best.pt source=./BVN.mp4 show=True
yolo detect predict model=runs/detect/train/weights/best.pt source=ultralytics/assets/aaa.jpg show=True