如何为训练暹罗神经网络而准备数据集

时间:2019-07-02 14:44:22

标签: python keras neural-network conv-neural-network

我正在尝试使用keras实现暹罗神经网络人脸识别模型。以前,我已经通过分类实现了CNN人脸识别模型(带有keras),因此我的数据集已经分为训练数据,训练标签,验证数据,验证标签和测试数据。我为此目的使用的数据集包含226个图像类。我用2799训练和226图像进行验证。现在我想使用数据集来训练暹罗神经网络。我该如何拆分数据?

1 个答案:

答案 0 :(得分:0)

对于暹罗神经网络,每个班级需要的图像数量要少得多。这些网络学会区分而不是分类。如果您从每个班级中提取一幅图片,表示226个班级,那么:

  • 形成图像对。如果一对中的两个图像都属于同一类,则关联的标签为1,否则将其设置为0。因此,最大对数非常大(226 ^ 226)。将这些对限制为一定数量。

请参阅此blog