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