当前位置:   article > 正文

yolov8多分支任务头训练

yolov8多分支任务头训练

目前已知的yolov8可以针对多个任务进行单独训练,但是暂时还没有开放针对多个任务头同时进行训练的教程,本文章针对yolov8的多任务训练进行详细介绍。
先放上效果图:
在这里插入图片描述
三个任务,分别是目标检测、可行驶区域、车道线,具体步骤请往下看。

一、环境配置
从如下github下载代码:

https://github.com/JiayuanWang-JW/YOLOv8-multi-task
然后自行根据原yolov8的requirement.txt配置虚拟环境。
二、训练数据配置
**2.1、**首先,如下图所示,按照GitHub中所说的数据格式进行配置,当然,数据有点多,我将自行挑选的100张图数据放置如下百度网盘,还有预训练权重,有需要请自行下载:
链接:https://pan.baidu.com/s/1O3W0T_xdNFv5gfLaXo0Ddw?pwd=1234
提取码:1234
–来自百度网盘超级会员V3的分享
在这里插入图片描述
**2.2、**然后,修改ultralytics路径下的train.py中数据路径为自己的数据路径:

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/不正经/article/detail/373518
推荐阅读
相关标签
  

闽ICP备14008679号