我如何微调ResNet50 Keras只分类2类(猫与狗)中的图像而不是所有1000个imagenet类?

时间:2017-12-07 00:42:02

标签: python classification imagenet

如何微调ResNet50 Keras只分类2类(猫与狗)中的图像而不是所有1000个图像网类?我使用Python并能够使用ResNet50和keras将随机图像分类为1000个ImageNet类。现在我想微调我的代码,以便它只使用Kaggle图像而不是ImageNet对猫与狗进行分类。我该怎么做呢?

1 个答案:

答案 0 :(得分:0)

有几种方法可以应用转学习及其试用版和试用版。什么效果最好的错误。然而,ImageNet在其1000个类别中包括多种类型的猫和狗,这就是为什么我会做以下事情:

  1. 将单个Dense图层添加到具有2个输出的模型
  2. 仅将最后一层设为可训练
  3. 仅使用猫狗图像重新训练网络
  4. 这将很快获得可靠的结果,因为您只训练一层。这意味着,您不必在整个网络中进行反向传播。此外,您的模型只需要学习从原始猫和狗子类到此二进制输出的相当线性映射。