温馨提示×

Keras中怎么实现迁移学习

小亿
84
2024-03-14 14:01:29
栏目: 深度学习

要在Keras中实现迁移学习,可以使用预训练的模型作为基础模型,然后在新的数据集上微调模型。以下是一个基本的迁移学习示例:

  1. 加载预训练的模型(例如ResNet50)
from keras.applications import ResNet50

base_model = ResNet50(weights='imagenet', include_top=False, input_shape=(224, 224, 3))
  1. 冻结基础模型的所有层
for layer in base_model.layers:
    layer.trainable = False
  1. 在基础模型的顶部添加自定义的全连接层
from keras.models import Model
from keras.layers import Dense, GlobalAveragePooling2D

x = base_model.output
x = GlobalAveragePooling2D()(x)
x = Dense(1024, activation='relu')(x)
predictions = Dense(num_classes, activation='softmax')(x)

model = Model(inputs=base_model.input, outputs=predictions)
  1. 编译模型并训练
model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
model.fit(train_data, train_labels, epochs=10, batch_size=32)

在这个示例中,我们使用ResNet50作为基础模型,冻结了所有的层,并在顶部添加了自定义的全连接层。然后编译模型并在新的数据集上训练模型。通过这种方式,我们可以利用预训练模型的特征提取能力,并在新的任务上进行微调。

0