分类：冻结网络层

2020
10-08

Keras 实现加载预训练模型并冻结网络的层

在解决一个任务时，我会选择加载预训练模型并逐步fine-tune。比如，分类任务中，优异的深度学习网络有很多。ResNet,VGG,Xception等等...并且这些模型参数已经在imagenet数据集中训练的很好了，可以直接拿过来用。根据自己的任务，训练一下最后的分类层即可得到比较好的结果。此时，就需要“冻结”预训练模型的所有层，即这些层的权重永不会更新。以Xception为例：加载预训练模型：fromtensorflow.python.keras.applicationsim... 继续阅读 >