深度学习-第二章-Window下Yolov5训练打电话

下载源码

github：git clone https://github.com/ultralytics/yolov5

并用vscode打开；注意切换到V6.0的分支

新建data目录

# 新建备份data
cp -rp data data_old

准备训练集

将img.zip拷贝到刚创建的data目录下，并解压
同时拷贝custome.yaml到此文件夹

有图片和对应的label信息

修改配置

修改custom.yaml中类别

拷贝process-date文件夹到data目录

修改yolov5s.yaml设置类别

在根目录新建weights权重文件

mkdir weights

在地址https://github.com/ultralytics/yolov5/releases下载yolov5s.pt文件。

放入weights目录

修改train.py中自定义部分

parser.add_argument('--weights', type=str, default=</