datagen.flow_from_directory()怎么将图片标记为子文件名
时间: 2024-05-16 15:12:50 浏览: 12
在使用`datagen.flow_from_directory()`函数时,可以通过设置参数`class_mode`为`'categorical'`,将图片标记为子文件名。具体实现方法如下:
1. 将图片按照类别分别存放在子文件夹中,每个子文件夹的名称即为对应类别的名称。
2. 使用`datagen.flow_from_directory()`函数读取图片数据,并将参数`class_mode`设置为`'categorical'`。
3. 读取得到的`ImageDataGenerator`对象中的`class_indices`属性即可得到每个类别的索引值。
4. 可以通过`class_indices`属性将索引值与类别名称进行对应,并得到一个字典对象,即`{类别名称: 索引值}`。
下面是一个示例代码:
``` python
from tensorflow.keras.preprocessing.image import ImageDataGenerator
# 数据增强生成器
datagen = ImageDataGenerator(
rescale=1./255,
rotation_range=20,
width_shift_range=0.2,
height_shift_range=0.2,
shear_range=0.2,
zoom_range=0.2,
horizontal_flip=True,
fill_mode='nearest')
# 从目录中读取图片数据
train_generator = datagen.flow_from_directory(
'dataset/train',
target_size=(224, 224),
batch_size=32,
class_mode='categorical')
# 获取每个类别的索引值
class_indices = train_generator.class_indices
print(class_indices) # {'cat': 0, 'dog': 1}
# 将索引值与类别名称进行对应
labels = {v: k for k, v in class_indices.items()}
print(labels) # {0: 'cat', 1: 'dog'}
```
运行以上代码,可以得到每个类别的索引值和对应的类别名称。其中`class_indices`属性得到的是一个字典对象,键为类别名称,值为对应的索引值;`labels`字典对象则是将索引值与类别名称进行了对应。
相关推荐
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![rar](https://img-home.csdnimg.cn/images/20210720083606.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)