使用ImageDataGenerator对象的flow_from_directory方法,创建一个生成器对象,哪个命名参数指定图像大小
时间: 2024-03-16 17:47:27 浏览: 104
图片大小生成器
4星 · 用户满意度95%
命名参数是target_size,它指定了生成器输出的图像的大小。可以将其设置为一个元组,如(height,width),也可以将其设置为一个整数,表示高度和宽度相等的正方形图像。例如:
```python
from keras.preprocessing.image import ImageDataGenerator
datagen = ImageDataGenerator(rescale=1./255)
generator = datagen.flow_from_directory(
directory='data',
target_size=(224, 224),
batch_size=32,
class_mode='binary'
)
```
这里的`target_size`设置为`(224, 224)`,表示生成器输出的图像大小为 $224\times224$。
阅读全文