YOLOv5ds搭建流程及遇到的bug

1、环境配置

git clone https://github.com/midasklr/yolov5ds.git
conda create -n yolov5ds
Conda activate yolov5ds
pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html
pip install -r requirements.txt

2、下载预训练模型（可略，代码会自动下载）

在yolov5ds-main根目录新建weights文件夹
下载yolov5预训练模型Releases · ultralytics/yolov5 · GitHub放到weights文件夹中

3、准备数据集——非常关键

det文件夹存放检测数据集yolo格式
seg文件夹存放分割数据集 labels存放分割图像.png

4、配置文件参数修改

4-1、models/segheads.yaml
segnc：改为自己的分割类别数 + 1（这里一定要+1）

4-2、data/voc.yaml
train：改为自己det文件夹下train.txt路径
val：改为自己det文件夹下val.txt路径
road_seg_t