Keras 图片分类算法中的参数设置简介
数据增强(Data Augment)
一般只对训练集增强,不对验证集增强。
1 | ImageDataGenerator.flow_from_directory(TRAIN_PATH, class_mode='categorical', #one hot encode |
class_mode=’categorical’: 使用one-hot编码
class_mode=’binary’: 使用二进制编码,一般用于二分类
target_size: 用来调整网络输入图片大小
batch_size:一次生成图片数