培训和预测时,Keras Feature Wise中心

时间:2018-08-15 18:09:29

标签: keras mean standard-deviation image-preprocessing

我已经使用imagewiseGenerator实现了featurewise_center = True,featurewise_std_normalization = True。我将生成器安装到我的训练集中(它会学习一些统计信息)并训练我的模型。一切都很好。

训练结束并关闭python之后,如何使用生成器对测试集进行相同的预处理?我想保留从拟合中学到的统计信息,以便在我要测试或进一步训练模型时使用(在这种情况下,原始训练集可能不再可用)。

预先感谢

1 个答案:

答案 0 :(得分:0)

我已经解决了这个问题。对于那些仍然遇到它的人。请按照以下步骤操作。

  1. 使用所需的预处理创建ImageDataGenerator。
  2. 从训练数据中随机抽取0.1%(即样本)。加载到内存中 适合ImageDataGenerator
  3. 从ImageDataGenerator获取get Mean和STD并将其放在列表中
  4. 对数据集进行多次迭代,并在每次迭代中保存均值和标准差
  5. 计算平均均值和STD
  6. 列表项

将平均平均值和平均STD分配给新的ImageDataGenerator。