Keras模型训练小技巧

Keras模型训练小技巧

  1. 建立基线模型
    首先建立一个模型,以训练以后的效果做参考,持续改进。
  2. 模型正则化
    a. dropout
    b. 数据增强
  3. 提示训练数据偏差
  4. 迁移学习
    a. 使用已训练的模型(加载预训练权重)
    b. 重新构建顶层
    c. 锁定其他层
    d. 重新训练
    e. 保存新训练的模型信息
    f. 加载保存的模型信息
    e. 对更多顶层进行训练
  5. VGG网络对输入图片进行color center处理(减去bgr均值(123.68, 116.779, 103.939))