keras同时使用迁移学习和函数api
# build model
base_model = ResNet18(input_shape=(224,224,3), weights='imagenet', include_top=False)
x = keras.layers.GlobalAveragePooling2D()(base_model.output)
output = keras.layers.Dense(n_classes, ...
原创
2020-02-19 23:30:47 ·
155 阅读 ·
0 评论