Tensorflow:微调初始模型

时间:2016-10-13 09:17:35

标签: python machine-learning tensorflow deep-learning

几天后,我按照此处的说明操作:https://github.com/tensorflow/models/tree/master/inception 用于微调初始模型。问题是我的数据集很大,因此将其转换为TFRecords格式将填满我的整个硬盘空间。有没有使用这种格式进行微调的方法?谢谢!

2 个答案:

答案 0 :(得分:1)

微调与数据格式无关;你在那里很好。 TFRecords促进训练和得分速度;它不应该影响所需的迭代次数或时期,也不应影响最终的分类准确性。

答案 1 :(得分:1)

您可以在不将数据转换为tfrecords的情况下训练任何模型。 Here通过直接从jpg文件中读取,可以很好地调整VGG。您可以将苗条的架构更改为Inception one,您应该没问题!

在当前配置中,必须将数据集划分为train和test文件夹,并将类作为子文件夹。但你可以把它改成你想要的任何东西。

与使用tfrecords的代码相比,我没有遇到任何代码速度的巨大差异。

相关问题