train_datagen.flow_from_directory中的train_dir 如何定义
时间: 2023-08-15 08:03:20 浏览: 89
python脚本生成caffe train_list.txt的方法
train_dir是指训练集所在的文件夹路径,通常该文件夹下会有多个子文件夹,每个子文件夹下都存放着同一类别的图片。在使用train_datagen.flow_from_directory时,需要将该路径传入函数中作为参数,示例代码如下:
```python
train_datagen = ImageDataGenerator(...)
train_generator = train_datagen.flow_from_directory(
train_dir,
target_size=(img_height, img_width),
batch_size=batch_size,
class_mode='categorical')
```
其中,train_dir应该是一个字符串类型的变量,指向训练集所在的文件夹路径。
阅读全文