YOLOV8从零搭建一套目标检测系统（修改model结构必看）附一份工业缺陷检测数据集

目录

1.YOLOV8介绍

2.YOLOV8安装

2.1环境配置

3.数据集准备

---

1.YOLOV8介绍

Yolov8结构图： 

YoloV8相对于YoloV5的改进点：
Replace the C3 module with the C2f module.
Replace the first 6x6 Conv with 3x3 Conv in the Backbone.
Delete two Convs (No.10 and No.14 in the YOLOv5 config).
Replace the first 1x1 Conv with 3x3 Conv in the Bottleneck.
Use decoupled head and delete the objectness branch.
anchor free.
 

YoloV8精度对比： 

 

2.YOLOV8安装

如果只是为了单纯使用，看下面这篇文章就好。

YOLOv8 保姆级教程（训练自己的数据集）_陈子迩的博客-CSDN博客

如果想为了改变模型的结构，接着往下看。

2.1环境配置

首先推荐使用anaconda配置python环境，不会的看下面这篇博客

最新Anaconda安装-保姆级教程_陈子迩的博客-CSDN博客

conda create -n YOLOv8 python=3.8  #创建YOLOv8的环境
 
conda activate YOLOv8   #激活环境
 
 

安装pytorch

CUDA 11.6
pip install torch==1.12.0+cu116 torchvision==0.13.0+cu116 --extra-index-url https://download.pytorch.org/whl/cu116 -i https://pypi.tuna.tsinghua.edu.cn/simple
CUDA 11.3
pip install torch==1.12.0+cu113 torchvision==0.13.0+cu113 --extra-index-url https://download.pytorch.org/whl/cu113 -i https://pypi.tuna.tsinghua.edu.cn/simple

     任选其一   

安装依赖包

pip install -r requirements.txt -i https://pypi.mirrors.ustc.edu.cn/simple/

打开pycharm，载入你的环境。

 

 选择你的python版本，正常新建的conda解释器会自动弹出

接下来 打开pycharm的终端，就是最底下的一行

选择这一个 

如果你安装了 ultralytics 包 ，那请卸载掉，如果没有装，跳过这一步。

运行python setup.py install.

 不用管中间出现啥

怎么判断自己是否安装成功，主要是看最后输出是否有Finished processing dependencies for ultralytics即可.

3.数据集准备

链接：https://pan.baidu.com/s/1FaBTUQvceUJJu3s1dg4xMg 
提取码：ypwa

给大家准备了一个钢铁的数据集

data.yaml放到下面路径

 数据集放在下面路径即可

 修改yaml中的地址

 创建一个my train的文件

 放入下面代码

from ultralytics import YOLO

# model = YOLO('yolov8n-CF2_ATT.yaml')
# model.train(data='data.yaml', epochs=5)

model = YOLO('yolov8n.yaml')
model.train(data='data.yaml', epochs=10)

 model.train函数的参数在下面选择

 配置好后就可以训练了，你也可以根据自己的需求修改模型配置文件

查看结果

配置my val

from ultralytics import YOLO

model = YOLO(r'D:\YOLOv8\ultralytics\models\yolo\detect\runs\detect\train11\weights\best.pt')
# It'll use the data yaml file in model.pt if you don't set data.
model.val()
# or you can set the data you want to val
model.val(data='data.yaml')

 以上就是完整的流程

yolov8可能遇到的报错下篇文章给大家罗列