datagen.flow_from_directory()怎么获取读取图片的名称
时间: 2024-05-08 14:22:27 浏览: 123
获取某一路径下所有图片名称
使用`flow_from_directory()`方法读取图片时,可以通过设置`class_mode`参数为`'categorical'`或`'binary'`来返回图片的标签信息,但是不会返回图片的文件名信息。
如果需要获取图片的文件名,可以在调用`flow_from_directory()`方法时设置`shuffle=False`,然后获取生成器的`filenames`属性即可,例如:
```python
from keras.preprocessing.image import ImageDataGenerator
datagen = ImageDataGenerator()
generator = datagen.flow_from_directory(directory='path/to/directory', shuffle=False)
filenames = generator.filenames
```
这样就可以获取到读取的每张图片的文件名了。
阅读全文