物体检测-系列教程9：YOLOV5 训练教程

1 训练

在prompt中(要cd到yolov5文件夹)

python train.py --data data.yaml --epochs 300 --weights '' --cfg yolov5s.yaml --batch-size 16

在IDE中填写运行配置参数

--data data.yaml --epochs 300 --weights '' --cfg yolov5s.yaml --batch-size 16

2 检测

启动摄像头测试

python detect.py --source 0
python detect.py --source 0 --weights ya.pt
python detect.py --source 1 --weights ya.pt

3 关于numpy的bug

yolov5的新版本中，在requirements中默认安装了numpy1.24，在numpy1.24中已经删除了numpy.int

建议：卸载新版本，安装老版本

pip uninstall numpy
pip install numpy==1.22

不建议：

改代码，将所有numpy.int改为numpy.int_

4 关于wandb的bug

建议:直接卸载wandb，不要安装即可

不建议：改代码

5 关于cudnn的bug

原因是有的代码会默认将一部分东西放到0号显卡去，暂时没搞懂什么样的代码会

这个时候，由于0号显卡满了，就会报上述错误

在train.py文件中加上这些代码，光速解决

import os
os.environ['CUDA_VISIBLE_DEVICES'] = "1,2,3"

6 关于GPU的bug

有时候会提示你‘页面太大，无法完成操作’，有时候会提示'RuntimeError: CUDA out of memory'

你只需要修改这个代码，就可以光速解决问题

parser.add_argument('--workers', type=int, default=8, help='maximum number of dataloader workers')

这里default=8，改成0即可

这个default=8表示的是在linux环境中运行