繁体   English   中英

Tensorflow:微调 Inception 模型

[英]Tensorflow: Fine tune Inception model

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

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

您可以训练任何模型而无需将数据转换为 tfrecords。 这里有一个很好的要点,可以通过直接从 jpg 文件中读取来微调 VGG。 您可以将纤薄架构更改为 Inception 架构,您应该没问题!

在当前配置中,您的数据集必须分为训练和测试文件夹,类作为子文件夹。 但是您可以将其更改为您想要的任何内容。

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

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM