当前位置: 代码网 > 科技>人工智能>车联网 > YOLOV8从零搭建一套目标检测系统(修改model结构必看)附一份工业缺陷检测数据集

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

2024年07月28日 车联网 我要评论
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.

目录

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可能遇到的报错下篇文章给大家罗列

(0)

相关文章:

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论

验证码:
Copyright © 2017-2025  代码网 保留所有权利. 粤ICP备2024248653号
站长QQ:2386932994 | 联系邮箱:2386932994@qq.com