train_gen = train_images.flow_from_directory(directory=train_dir, batch_size=batch_size, shuffle=True,#随机载入 target_size=(im_size, im_size),#修改图片大小 class_mode='categorical')#标签转换成这段代码是什么意思
时间: 2024-02-10 11:18:54 浏览: 68
Keras之fit_generator与train_on_batch用法
这段代码是用来创建一个图像数据生成器,可以方便地从指定的目录中加载图像数据进行训练。其中:
- `directory`参数指定了加载数据的目录;
- `batch_size`参数指定了每个batch中包含的样本数;
- `shuffle`参数指定了是否对样本进行随机打乱;
- `target_size`参数指定了将图像调整为的大小;
- `class_mode`参数指定了标签的类型,这里设置为`categorical`表示采用one-hot编码的形式。
阅读全文