我的数据集中有四个类别(A、B、C、D)。我不是使用扁平的多类分类器,而是尝试使用多个二元分类器。例如:
首先 – A vs B
然后(A和B vs C)
再然后(A、B和C vs D)
如何通过使用flow_from_directory来实现这一点,而不必每次都将文件夹的内容合并到新文件夹中?
谢谢
回答:
对于第一种情况,你可以指定flow_from_directory中的classes
属性
generator.flow_from_directory('./train', classes=['A', 'B'])
对于其他情况,请查看这个答案 https://stackoverflow.com/a/52392763/844372