keras中的base_model
1、导包
from tensorflow.keras.applications import VGG16
2、基本模型
1 img_size = 224
2 base_model = VGG16(include_top=False, # 是否保留顶层的3个全连接网络
3 weights='imagenet', # 是否加载预训练权重
4 input_shape=(img_size,img_size,3))
3、参数
include_top:是否保留顶层,默认为True
input_shape:当include_top为False时需要设置输入图像的大小
weights:是否加载预训练权重