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')