批处理如何同时包含正面和负面标签？

Question

我有要使用VGG-16解决的行人分类问题。为此，我准备了训练和测试仪。我的火车设置了2038张图像，我的测试设置了252张图像。我的批量大小是64。如何在训练时告诉Keras我希望这64张图像同时包含正负标签？我不希望它只在正面或负面标签上学习。

Answer 1

如果您改组训练数据（强烈建议使用），则获得属于同一类别的所有64个样本的机会很小。对于两类任务，数据应该没有问题。

但是，如果您希望保证培训批次的平衡，则可以使用BalancedBatchGenerator之类的第三方代码。有关更多信息，请参见此tutorial。