Keras实现迁移学习

 

用ImageNet中VGG16的权重训练自己的分类模型：

1. 先用Keras的已拥有的model.load_weights(best_model_file)的方法把之前训练好的500分类的模型的权重先加载进来，现在的模型是就的500分类的模型；
2. 最后一层的全连接层里在原有的500分类的权重基础上，再在张量矩阵里面加上N个新添加分类数的随机数矩阵。这样子就可以使用之前已经训练好的500分类的权重，又可以在新的训练里面添加上新的分类数。
3. 第三步把新的全连接放到Model的outputs里面创造一个全新的模型作为训练使用。