用yolo_v2训练自己的数据,,参考各类博客,准备好了数据,修改了配置文件,但训练时不断出现各种问题。刚开始出现loss为nan的情况,经排查是学习速率太大,loss发散了(因训练数据较少,约1500张图片)
再次训练出现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。学习速率按自己的实际情况调节,建议慢慢试(太大太小都不行,太大发散,太小易陷入局部最优)
成功解决~
yolo_v2训练自己的数据出现Obj: 0.000000, No Obj: 0.000000
最新推荐文章于 2024-04-20 11:35:34 发布