再培训一门二进制图像分类器(猫/狗)

时间:2019-04-22 07:29:39

标签: image multiclass-classification

我正在使用以下代码:

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,但这似乎不是问题。

我希望模型的精度能够像基础模型那样不断提高。

0 个答案:

没有答案