BryanNCAG 2022-05-30 11:33 采纳率: 0%
浏览 655

YOLOv5训练mAP值上不去

设备1:主机
i5-6600T
翔升GTX 1650
驱动程序版本471.11
Cuda版本11.4
Cudnn版本8.4.0.27

设备2:笔记本yoga 14s
R7-4800U
无独显

训练对象:
类似于路牌上,标准蓝色背景的向左和向右的箭头,统一归类为横向箭头(class name),只有这里一个class。

问题描述和横向对比:

  1. 使用yoga 14s,调用CPU
    1-1. 用现有模型检测,效果OK
    1-2. 用CPU训练新模型
    mAP值随epoch正常提升,起初为0.0001左右,10个epoch后到0.5以上,最终结束于0.75~0.8。模型后续实测效果良好。
  2. 使用主机,调用GPU
    2-1. 用现有模型检测,效果OK
    2-2. 用GPU训练新模型
    mAP值始终在低位徘徊,起初为10^(-8),后续一直在10^(-8)~10^(-5)徘徊。

训练使用相同数据集和train的参数:
--img 640
--batch 4
--epochs 100
--workers 0

目前的尝试解决方案:
显卡驱动更新,从原来的456多,更新到471.11,本希望满足pytorch官网上所述的对应11.3版本cudatoolkit的pytorch,但是studio驱动只找到471.11,因此现在的cuda是11.4的版本。

  • 写回答

2条回答 默认 最新

  • 对象被抛出 2022-05-30 11:47
    关注

    指标上不去和驱动, 环境没关系, 虽然yolo确实是做检测的, 但不是说你拿过来什么都不改就能在一个新的数据上有很好的表现的

    评论

报告相同问题?

问题事件

  • 修改了问题 5月30日
  • 修改了问题 5月30日
  • 创建了问题 5月30日

悬赏问题

  • ¥20 用户端如何上传图片到服务器和数据库里
  • ¥15 现在研究生在烦开题,看了一些文献,但不知道自己要做什么,求指导。
  • ¥15 vivado封装时总是显示缺少一个dcp文件
  • ¥100 pxe uefi启动 tinycore
  • ¥15 我pycharm运行jupyter时出现Jupyter server process exited with code 1,然后打开cmd显示如下
  • ¥15 可否使用carsim-simulink进行四轮独立转向汽车的联合仿真,实现四轮独立转向汽车原地旋转、斜向形式、横移等动作,如果可以的话在carsim中如何进行相应设置
  • ¥15 Caché 2016 在Java环境通过jdbc 执行sql报Parameter list mismatch错误,但是同样的sql使用连接工具可以查询出数据
  • ¥15 疾病的获得与年龄是否有关
  • ¥15 opencv.js内存,CPU飙升
  • ¥15 植物重测序snp数据Treemix分析出现问题!