现象说明

目标:二分类,训练过程中验证集准确率比较高。训练完成之后对测试集进行测试,测试结果中的类别全部为1(最后一类)。
Caffe测试集分类结果相同,全部为最后一类

原因及解决方案

原因:没有对网络的第一个卷积层进行初始化。下图是原来网络第一个卷积层的参数。
Caffe测试集分类结果相同,全部为最后一类
解决方案:将第一个卷积层权重初始化为“xavier”,偏置初始化为常量0。
Caffe测试集分类结果相同,全部为最后一类

相关文章:

  • 2022-12-23
  • 2021-12-19
  • 2022-12-23
  • 2021-12-13
  • 2021-09-03
猜你喜欢
  • 2021-09-15
  • 2022-01-07
  • 2021-12-18
  • 2022-01-05
  • 2022-12-23
  • 2021-12-11
  • 2021-07-15
相关资源
相似解决方案