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

阿尔法萨尔·阿尔巴克里(Alfaisal Albakri)

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

Shubham Panchal

对于暹罗NN,每个类需要的图像数量要少得多。这些网络学会区分而不是分类。如果您从每个班级中提取一个单一的意思是226个班级,则:

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

请参阅此博客

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章