2

如果我有这样的目录结构:

|-data
   |-train
      |-Bellis perennis
      |   |-lawn_daisy
      |   |     |-image1.jpg
      |   |-english_daisy
      |         |-image1.jpg
      |-Tulipa dasystemon
      |   |-some tulip name
      |         |-image.jpg
      |-etc

我使用 Keras flow_from_directory,它只会抓取第一棵树,即 Bellis perennis 和 Tuplia 并将它们用作标签。有没有办法将主目录(Bellis)用作一个标签,将其子目录(草坪雏菊,english_daisy)用作第二个标签,并将这些标签与 Keras 功能 API 一起使用?

另一种选择是仅更改文件夹的设置方式,但如果有办法使用此文件夹结构,我想了解它。

编辑 - 进一步阅读后,您似乎可以在train_datagen.flow_from_directory. 但是你能指定2个不同的类吗?

4

0 回答 0