小白也能轻松上手!YOLOv8全流程教程,毕设必备!
一、PyCharm安装(保姆级)
首先,我们得有个好帮手——PyCharm。下载免费版或专业版,安装过程简单,记得勾选“Add to PATH”和“Create Desktop Shortcut”哦。
1. 下载与安装
访问官网下载,选择Community(免费)或Professional版本。
2. 配置Python环境(conda创建虚拟环境)
在终端输入以下指令创建虚拟环境:
conda create -n yolov8 python=3.8
然后激活虚拟环境:
conda activate yolov8
安装YOLOv8库:
pip install ultralytics==8.0.73
3. 下载yolov8源码,在pycharm中进行配置
从GitHub下载源码,解压后用PyCharm打开,设置interpreter为创建的虚拟环境。
from ultralytics import YOLO
model = YOLO('yolov8n.pt') # 加载官方预训练模型
model.info() # 查看模型信息
二、GPU环境配置(Windows + NVIDIA CUDA)
1. 检查GPU
在命令行输入:
nvidia-smi
如果显示显卡信息,说明GPU驱动正常。
2. 安装CUDA & cuDNN
参考之前的博客安装CUDA和cuDNN,并配置系统环境变量。
3. 安装GPU版PyTorch
访问PyTorch官网,选择CUDA版本,复制命令安装。
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.8 -c pytorch -c nvidia
三、LabelImg标注(标注自己数据集)
1. 安装LabelImg
使用pip安装:
pip install labelimg -i https://pypi.tuna.tsinghua.edu.cn/simple
2. 启动LabelImg
在cmd中输入:
labelimg
选择图片文件夹,标注目标,保存为VOC格式。
3. 转换为YOLO格式
如果标注时选择的是YOLO格式,则无需转换。
4. 整理数据集结构
YOLOv8数据集推荐结构:
dataset/
├── images/
│ ├── train/
│ └── val/
├── labels/
│ ├── train/
│ └── val/
四、自定义YOLOv8数据集训练
1. 创建数据集YAML文件
dataset.yaml
train: dataset/images/train
val: dataset/images/val
nc: 3 # 类别数
names: ['cat', 'dog', 'person'] # 类别名称
2. 开始训练
使用以下命令开始训练:
yolo detect train data=dataset.yaml model=yolov8n.pt epochs=100 imgsz=640 batch=16 device=0
3. 查看训练效果
训练完成后,PyTorch会生成runs/detect/train文件夹,其中包含模型权重、训练曲线和推理结果。
五、模型推理(验证效果)
from ultralytics import YOLO
model = YOLO('runs/detect/train/weights/best.pt')
model.predict('test.jpg', show=True) # 在图片上显示预测结果
六、小贴士 & 注意事项
- GPU显存小 → 使用yolov8n.pt或减小batch
- LabelMe标注类别统一,文件名不要有中文
- Windows用户注意路径分隔符,最好用/
- 训练前确认CUDA与PyTorch对应版本匹配
七、总结
本文覆盖了从PyCharm安装、GPU配置、LabelMe标注到YOLOv8自定义训练的完整流程,小白也可以快速上手毕设项目。掌握这套流程,你就能在自己的数据集上进行目标检测和实例分割训练。
