我正在使用以下代码:
https://github.com/llSourcell/how_to_make_an_image_classifier/blob/master/demo.ipynb
他在那里做一个简单的猫与狗的比较。我使用了自己的图像集,该图像集仅包含240个训练图像和50个测试样本,并且尝试了相同的模型。我在模型中唯一更改的是这个
img_width,img_height = 432,288
nb_train_samples = 720 nb_validation_samples = 150
对于两个类别,在五个纪元内,它的好坏都达到了100%的验证准确性。当我在模型/火车/混合和模型/验证/混合中添加第三类时。然后,它给我的稳定精度仅为66%,并且从那里没有任何改善。我要去哪里错了?
对于代码:
https://colab.research.google.com/drive/1iMJCLsQDC73obejWsdAxygpvGAWMb2Gw
我尝试将此类纳入培训和验证。更改了class_mode,但这似乎不是问题。
我希望模型的精度能够像基础模型那样不断提高。