锐客网

  1. 首页 > 睿知 > it技术 > >

深度学习|自动驾驶－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
然后，会有一个提示网址，打开网页进入
深度学习|自动驾驶－YOLOV5网络训练（可能会遇到五个坑）
文章图片


注册登录后（如果有帐号，则直接登录即可），会生成一个KEY
深度学习|自动驾驶－YOLOV5网络训练（可能会遇到五个坑）
文章图片



在终端中输入（ctrl + v在终端中不会出现你粘贴的东西），粘贴后 enter 就好。
然后选择创建一个新的还是远来已经创建好的，就可以：
深度学习|自动驾驶－YOLOV5网络训练（可能会遇到五个坑）
文章图片



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中配置文件，默认值为空造成：改成配置如下图即可
深度学习|自动驾驶－YOLOV5网络训练（可能会遇到五个坑）
文章图片

如果看不懂的话，可参阅另一朋友的详细篇
地址：RuntimeError: Given groups=1, weight of size [512, 1024, 1, 1], expected input[1, 512, 8, 8] to have_诺顿与苏菲的博客-CSDN博客
8、本人的运行报告如下。
深度学习|自动驾驶－YOLOV5网络训练（可能会遇到五个坑）
文章图片

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

完


【深度学习|自动驾驶－YOLOV5网络训练（可能会遇到五个坑）】

    推荐阅读


    上一篇：#|【计算机视觉】计算机视觉与深度学习-北邮鲁鹏老师课程笔记

    下一篇：单阶段目标检测|YOLOv5-v6.0-网络架构详解（第二篇）