跳转到主内容
思享编程网:思考分享,玩转编程世界!

小白也能轻松上手!YOLOv8全流程教程,毕设必备!

小白也能轻松上手!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自定义训练的完整流程,小白也可以快速上手毕设项目。掌握这套流程,你就能在自己的数据集上进行目标检测和实例分割训练。

我是陆砚码,来自思享编程网(www.sxgpb.com),如果你对编程技术感兴趣,欢迎关注我们的网站,获取更多精彩内容!

相关文章