当前位置: 首页 > news >正文

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

http://www.lryc.cn/news/159545.html

相关文章:

  • Maven 的其它插件
  • 系列十三、Java操作RocketMQ之带Key的消息
  • C#调用Dapper
  • 2023高教杯数学建模1:ABC题目+初步想法
  • ApachePulsar原理解析与应用实践(学习笔记一)
  • 2023开学礼《乡村振兴战略下传统村落文化旅游设计》许少辉八一新书南京财经大学图书馆
  • qt 信号与槽机制,登录界面跳转
  • uniapp的两个跳转方式
  • 【LeetCode】1654:到家的最少跳跃次数的解题思路 关于力扣无法return的BUG的讨论
  • Calico IP In IP模拟组网
  • 在linux上挂载windows共享目录
  • drone的简单使用
  • day 52 | 84.柱状图中最大的矩形
  • BUUCTF刷题十一道(08)
  • 快速构建基于Paddle Serving部署的Paddle Detection目标检测Docker镜像
  • SOLIDWORKS工程图自动零件序号的极致体验
  • 将ROS bag转成CSV
  • jframe生成柱状图片+图片垂直合并+钉钉机器人推送
  • 如何用J-Link仿真PY32F003系列芯片
  • # Go学习-Day10
  • vue3:5、组合式API-reactive和ref函数
  • Unity Inspector面板上显示Api
  • Redis功能实战篇之附近商户
  • selenium 自动化测试——元素定位
  • 【JMeter】 二次开发插件开发 Dubbo 接口测试插件浅析
  • 手机SSL证书认证失败是什么意思?
  • PXE网络批量装机(centos7)
  • P1104 生日
  • 计算机网络复习大纲
  • Linux:进程(概念)