多光谱遥感分类（二）：VGG微调

继上篇。

CNN模型

使用VGG16预训练模型。

from keras.preprocessing.image import ImageDataGenerator
from keras.callbacks import TensorBoard
from keras.applications import VGG16
from keras import layers,models,optimizers



train_pic_gen=ImageDataGenerator(rescale=1./255,rotation_range=20,width_shift_range=0.2,height_shift_range=0.2,
                                 shear_range=0.2,zoom_range=0.5,horizontal_flip=True,fill_mode='nearest')
test_pic_gen=ImageDataGenerator(rescale=1./255)
train_flow=train_pic_gen.flow_from_directory("data/1_train",(128,128),batch_size=128,class_mode='categorical')
test_flow=test_pic_gen.flow_from_directory("data/1_test",(128,128),batch_size=128,class_mode='categorical')

# print(test_flow.class_indices)
### {'169': 0, '26': 1, '6': 2}

conv_base=VGG16(weights='imagenet'