python - 训练模型中的错误，其中训练图像位于类子文件夹中，而测试图像位于另一个文件夹中

Question

我正在尝试对图像进行二进制分类。存在的图像有两种类型，例如“A”类和“B”类我的文件夹的结构是：

data/
    train/
         class 'A' images folder/
               1.jpg
               2.jpg .......
         class 'B' images folder/
               1.jpg
               2.jpg .......
    test/
        only images without any labels
    csv file containing test images name and their label/
        1.jpg  - 'A'
        2.jpg  - 'B'

我正在运行以下命令，但它给出了错误

model.fit_generator(
  training_set,
  validation_data=test_set,
  epochs=10,
  steps_per_epoch=len(training_set),
  validation_steps=len(test_set)
)

错误是：ValueError: logits and labels must have the same shape ((None, 2) vs (None, 1)) 我是 keras 的新手，因此不知道如何解决此错误。提前致谢。