肯定有许多同学在更改网络结构遇到过map=0的问题,但尝试n多次都不知道错误究竟出在哪里。如图1所示。
图1 map=0
我在初次遇到这个问题时不知道浪费了几天的时间,这次特意来给大家避坑。
我这次是因为更改主干网络导致的,因为主干网络的更改,我没有使用预训练模型。导致模型中权重太过随机,在经过100epoch竟然仍未学习到一点东西。我尝试增加数据集解决问题,从1000张数据集增长到10000数据集。但还是没任何作用。我最后决定...放弃...
换成yolov4的主干网络CSPDacknet53,并从网上下载了一个voc预训练权重验证我的想法,果不其然,就是没有预训练权重的问题。
这里我建议非到万不得已还是不要更改主干网络的好,but也可能是我太水了,不会调参啥的。有经验的博友希望可以在评论区留言教教我...
如果必须更改主干网络建议参考以下博客。(5条消息) Imagenet图像分类训练总结(基于Tensorflow 2.0实现)_gzroy的博客-CSDN博客_tensorflow2图像分类训练