Керас для создания CNN - PullRequest
0 голосов
/ 04 мая 2019

Я работаю с Керасом над созданием собственной CNN для идентификации кошек и собак. Я создал папку под названием "кошки и собаки". Эта папка находится в том же каталоге, что и моя записная книжка jupyter ( "кошки и собаки" папка & Keras.ipynb записная книжка оба находятся в папке Anaconda3). Теперь дальнейшее дерево папок выглядит так:

кошки-и-собака /
| _______________ поезд /
| _______________ | _______ кошки /
| _______________________ | _____ 1.jpg
| _______________________ | _____ 2.jpg
| _______________________ | _____...
| _______________ | _______ собаки /
| _______________________ | _____ 21.jpg
| _______________________ | _____ 22.jpg
| _______________________ | _____...
| _______________ тест /
| _______________ | _______ кошки /
| _______________________ | _____ 41.jpg
| _______________________ | _____ 42.jpg
| _______________________ | _____...
| _______________ | _______ собаки /
| _______________________ | _____ 61.jpg
| _______________________ | _____ 62.jpg
| _______________________ | _____...
| _______________ Valid /
| _______________ | _______ кошки /
| _______________________ | _____ 81.jpg
| _______________________ | _____ 82.jpg
| _______________________ | _____...
| _______________ | _______ собаки /
| _______________________ | _____ 101.jpg
| _______________________ | _____ 102.jpg
| _______________________ | _____...
Keras.ipynb
| ____
| ____
...

Мой код, который выдает ошибку:

train_batches = ImageDataGenerator (). Flow_from_directory («кошки и собаки / поезд», target_size = (224,224), классы = ['dog', 'cat'], batch_size = 10)

Ожидаемый результат: Найдено 80 изображений, относящихся к 2 классам.
Фактический результат: Найдено 0 изображений, относящихся к 2 классам.

1 Ответ

1 голос
/ 04 мая 2019

Вы устанавливаете параметр classes на classes=['dog','cat'], это будет искать папки dog и cat, но ваши папки называются dogs и cats, поэтому имеет смысл, что он не находитimages.

Либо удалите параметр classes для генератора, чтобы автоматически определять имена классов, либо введите правильные имена классов, соответствующие именам папок.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...