Caffe测试集分类结果相同,全部为最后一类

现象说明 目标:二分类,训练过程中验证集准确率比较高。训练完成之后对测试集进行测试,测试结果中的类别全部为1(最后一类)。 Caffe测试集分类结果相同,全部为最后一类 原因及解决方案 原因:没有对网络的第一个卷积层进行初始化。下图是原来网络第一个卷积层的参数。 解决方案:将第一个卷积层权重初始化为“xavier”,偏置初始化为常量0。
相关文章
相关标签/搜索