YOLOv8-loopy

1.环境配置

• conda 安装

conda create -n yolov8-loopy python=3.12
conda activate yolov8-loopy

安装 ultralytics 时会自动安装 CPU 版本的 pytorch 和 torchvision，但我们一般需要安装 GPU 版本，因此我们可以先安装 pytorch 再安装 ultralytics 就可以避免这个问题。

根据自己的系统和 cuda 版本安装对应的 pytorch。

PyTorch 官网：PyTorch

最后再来安装 ultralytics

pip install ultralytics

2.训练

图像放在 dataset/images 下，划分为 train 和 val，标签放在 dataset/labels 下，和 images 一样划分

修改 yaml 文件

训练：

python train.py

模型保存在 runs/detect/train 下

3.推理

在 detcet.py 中设置好需要推理的文件和模型

python detect.py

输出文件在 runs/detect/predict 下