验证集准确率没变化
来源:6-10 10monkeys基础模型搭建与训练
Ams0
2019-10-01
老师,我运用了你写的这个分类10-monkeys的程序,来做其它的图像集分类。
我从kaggle上下载的猫狗图像来做二分类,是没有问题的,最后验证集的准确率有接近90左右,最后输出的loss和accuracy的曲线也是看上去没有问题的。
但是呢,我把图像集换成了另一组自己的图像集,也是作二分类,用同样的方式读取数据集,结果训练的时候却有些问题,就是验证集的loss不怎么变化,验证集准确率也不发生变化,总是保持一个值。
最后的激活函数用了softmax,损失函数categorical_crossentropy。也试过改成sigmoid 和 binary_crossentropy,结果都是这样。
不知道是不是过拟合?还是学习率的问题?
原图像是640*640的,我缩小到224 或 480,结果也还是这样,训练图像共233张,验证图像80张
,batch_size设置成10, 是cnn网络的层次参数等要改动吗?
以前用过tensorflow1.0来做过这个图像集的二分类,是没问题的,现在用keras来出现了这个问题,请问老师有啥指点吗?
写回答
1回答
-
正十七
2019-10-01
试试把learning_rate调低或者加上batch_normalization试一下。
00
相似问题