用yolo_v2训练自己的数据,,参考各类博客,准备好了数据,修改了配置文件,但训练时不断出现各种问题。刚开始出现loss为nan的情况,经排查是学习速率太大,loss发散了(因训练数据较少,约1500张图片)
再次训练出现Obj: 0.000000, No Obj: 0.000000的情况,这个问题说明网络没有检查到任何物体,截图如下:
yolo_v2训练自己的数据出现Obj: 0.000000, No Obj: 0.000000
解决步骤:
1.上网搜解决方案,有说是数据集问题,但检查了标签内容、存放路径、配置文件等,没问题
2.怀疑自己无意修改了yolo-xxx.cfg文件的某些参数,于是重新复制了yolo-voc.cfg的内容,修改batch=32,max_batches为120次epoch(即,训练数据量/32*120),最后一个[convolutional]中filters和[region]中的classes。学习速率按自己的实际情况调节,建议慢慢试(太大太小都不行,太大发散,太小易陷入局部最优)
成功解决~

相关文章:

  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2021-07-30
猜你喜欢
  • 2021-07-19
  • 2021-09-10
  • 2021-11-29
  • 2021-04-15
  • 2022-01-22
  • 2022-02-02
相关资源
相似解决方案