自动驾驶－YOLOV5网络训练（可能会遇到五个坑）

1、环境：

操作系统：Ubuntu20.04

Python3.9

Pytorch1.10.2

IDE：（本人使用的是ECLIPSE）

显卡：NVIDIA GeForce GTX 1060 6GB

2、网络训练

在YOLOV5目标下，选择：train.py，执行

3、可能会遇到的坑一

报：/bin/sh: pip: command not found

遇到这错误，说明环境不协调，需要安装配套的环境，具体每个人的环境不一样，需要细心解决，找资料。

4、可能会遇到的坑二

报：wandb.errors.UsageError: api_key not configured (no-tty). call wandb.login(key=[your_api_key])

这是需要在终端运行：

wandb init

然后，会有一个提示网址，打开网页进入

注册登录后（如果有帐号，则直接登录即可），会生成一个KEY

在终端中输入（ctrl + v  在终端中不会出现你粘贴的东西），粘贴后 enter 就好。

然后选择创建一个新的还是远来已经创建好的，就可以：

5、可能会遇到的坑三

报：RuntimeError: CUDA out of memory. Tried to allocate 2.8GB

这是说明显示显存，被其它程序占用，用清缓存的办法也行，但不是太干净。我的办法是直接重启了电脑，然后重新运行解决的。

6、可能会遇到的坑四

报：AttributeError: Cant get attribute SPPF on module models.common

遇到这个问题，说明你运行过YOLOV6，这是运行过YOLOV6后，运行V5才会有这问题

解决办法，请查阅另一个朋友的文档：

地址：运行yolov5出现问题AttributeError: Cant get attribute SPPF on module models.common_Steven_Cary的博客-CSDN博客

7、可能会遇到的坑五

报：RuntimeError: Given groups=1, weight of size [512, 1024, 1, 1], expected input[1, 512, 8, 8] to have

这个问题，是文件tran.py中配置文件，默认值为空造成：改成配置如下图即可

 如果看不懂的话，可参阅另一朋友的详细篇

地址：RuntimeError: Given groups=1, weight of size [512, 1024, 1, 1], expected input[1, 512, 8, 8] to have_诺顿与苏菲的博客-CSDN博客

8、本人的运行报告如下。

9、训练完后，得到模型： last.pt

这时，可以用自己训练获得的模型进行目标检测，把detect.py文档中，模型改成自己训练的模型即可（但本人训练的模型，由于没有精心调参训练的模型不如公开库里的模型）

完