目前已知的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中数据路径为自己的数据路径: