需求标注文件格式
xxx.jpg,0
xxx.jpg,1
xxx.jpg,2.......
抱歉,您需设置社区昵称后才能参与社区互动!
需求标注文件格式
xxx.jpg,0
xxx.jpg,1
xxx.jpg,2.......
|
发表于2020年02月23日 15:22:29
直达本楼层的链接
沙发
只看该作者
keras 可以直接使用ImageDataGenerator读取数据,DATA_PATH=yourpath/images/,classes需要和那个txt从上到下写个list对应一下就可以了 from keras.preprocessing.image import ImageDataGenerator train_gen = ImageDataGenerator( rotation_range=20,validation_split=0.3) train_generator = train_gen.flow_from_directory(DATA_PATH, target_size=(224, 224), classes=label_list, batch_size=batch_size, subset='training')# 验证数据 validation_generator = train_gen.flow_from_directory(DATA_PATH, target_size=(224, 224),
classes=label_list, batch_size=batch_size,
subset='validation') |
|
keras 可以直接使用ImageDataGenerator读取数据,DATA_PATH=yourpath/images/,classes需要和那个txt从上到下写个list对应一下就可以了
from keras.preprocessing.image import ImageDataGenerator train_gen = ImageDataGenerator( rotation_range=20,validation_split=0.3) train_generator = train_gen.flow_from_directory(DATA_PATH, target_size=(224, 224), classes=label_list, batch_size=batch_size, subset='training')# 验证数据
validation_generator = train_gen.flow_from_directory(DATA_PATH, target_size=(224, 224), classes=label_list, batch_size=batch_size, subset='validation')