文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

完整且详细的Yolov8复现+训练自己的数据集

2023-08-31 15:45

关注

Yolov8 的源代码下载:ultralytics/ultralytics: NEW - YOLOv8 🚀 in PyTorch > ONNX > CoreML > TFLite (github.com)https://github.com/ultralytics/ultralytics

Yolov8的权重下载:Releases · ultralytics/assets · GitHubUltralytics assets. Contribute to ultralytics/assets development by creating an account on GitHub.https://github.com/ultralytics/assets/releases

        yolov8做了更简单的部署,可以用于检测,分类,分割等,速度更快,精度更高。具体yolov8的复现可以参考:

一、代码,权重的下载

打开上面的源代码地址,下载源代码压缩包。

 2.下载后解压。

 

权重的下载 :建议点击上面的链接直接下载,后面的predict.py虽然设置了自动下载,但是往往因为网络或者环境配置的问题cut掉。因为作者复现的是检测任务,权重放在detect文件下。

二、配置环境

建议每次做新项目都重建一个新环境,避免了各种包的版本的冲突,同时也为了避免在新项目跑通后旧项目又要重新配置环境的麻烦,所以重建环境是不错的选择。

1.打开Anaconda Prompt(如果没有Anaconda,建议下载一个,在配置环境的方面还是很方便的,具体的下载方式参考:(54条消息) 史上最全最详细的Anaconda安装教程_OSurer的博客-CSDN博客https://blog.csdn.net/wq_ocean_/article/details/103889237?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522168653306416782427441050%2522%252C%2522scm%2522%253A%252220140713.130102334..%2522%257D&request_id=168653306416782427441050&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~top_positive~default-1-103889237-null-null.142^v88^control,239^v2^insert_chatgpt&utm_term=anaconda安装教程&spm=1018.2226.3001.4187

2.具体操作:

(1) 创建环境

conda create -n yolov8 python==3.7

(官方要求>=3.7,所以python3.8也完全可以)

(2)激活环境

conda activate yolov8

(3)下载Pytorch,这个步骤也十分重要!根据自己电脑配置下载。官方要求Pytorch>=1.7。

首先查看自己的显卡配置:win+R  ,输入nvidia-smi

 去官网下载对应或者不大于箭头指出的版本,官网地址:PyTorchhttps://pytorch.org/

pip下载会比conda下载略快。

(4)配置好环境后,使用pycharm打开源代码工程文件

(5)选择下载的yolov8环境。

 

选择python.exe文件。 

 (6)配置yolov8要求的包,点击Temina,输入代码:

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/

-i https://pypi.tuna.tsinghua.edu.cn/simple/这是一次性换源代码,要关掉梯子才能链接。

三、验证一下代码是否可以预测例子

打开predict.py

直接运行,结果会保存在runs里。

可能碰见的错误是关于torchvision版本的问题,重新安装即可。 

四、制作自己的数据集

先介绍YoLov 8 最终所需要的数据集格式:

datasets

        |-images

                |--train

                |--val

                |--test

        |-labels

                |--train

                |--val

                |--test

Yolo要求的数据标签为.txt

与Yolov7 和v5 一样,可以使用labelme标注数据集,yolov8支持多种数据集格式,我是采用上面的格式跑通了,具体制作的过程可参考:

(60条消息) YOLOv5系列 1、制作自己的数据集_yolov5数据集制作_冯璆鸣的博客-CSDN博客https://blog.csdn.net/fjlaym/article/details/123992962?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522168722415116800180654496%2522%252C%2522scm%2522%253A%252220140713.130102334..%2522%257D&request_id=168722415116800180654496&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~top_positive~default-1-123992962-null-null.142^v88^control,239^v2^insert_chatgpt&utm_term=yolov5制作自己的数据集&spm=1018.2226.3001.41873.制作好数据集后,建立.yaml文件,可建在任何位置,建议与v8其他yaml文件放在一起,修改路径就会方便很多。

 (1)修改模型配置文件

选择yolov8.yaml,修改nc为自己数据集所需检测类别的个数

 (2)修改数据加载配置文件,建议全部使用绝对路径

train: "D:/ultralytics-main/datasets/belt/train.txt"val: "D:/ultralytics-main/datasets/belt/train.txt"nc: 1names: ["1"]

至此,所有的配置已经完成。

五、训练自己的数据集

(1)yolo提供自己的指令模式,在调参方面十分方便,当然不下载也可以,直接在文件修改和运行也无碍。

在Terminal下直接运行:

 pip install ultralytics

(2)训练:

yolo train data=你的配置文件(xx.yaml)的绝对路径 model=yolov8s.pt epochs=300 imgsz=640 batch=8 workers=0 device=0

如果想使用多卡训练,device='\0,1,2,xxx\'

(3)训练过程首先会显示你所使用的训练的硬件设备信息,然后下一段话则是你的参数配置,紧接着是backbone信息,最后是加载信息,并告知你训练的结果会保存在runs\detect\trainxx。

来源地址:https://blog.csdn.net/chenhaogu/article/details/131161374

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     807人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     351人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     314人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     433人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     221人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯