使用 TF-Slim 在我自己的数据上微调预训练的 VGG16 模型

Question

我正在尝试使用 TF-Slim 微调 VGG-16 预训练模型，但是，我在开始时遇到了问题（尝试预处理数据时，即将图像从 tiff 转换为 jpeg，然后转换为 tfrecord 格式）。

谁能建议使用最佳方法来微调模型以适应这些图像？

或者也许可以为我指明如何执行此操作的分步指南？道歉，因为我是 convnets 的新手，我正试图弄清楚如何开始微调。

提前致谢。

Answer 1

您可以在下面关于花卉数据集的 TF slim 演练中获得有关如何微调预训练模型的详细步骤 https://github.com/tensorflow/models/blob/master/research/slim/slim_walkthrough.ipynb

另外，我建议您查看下面的 git 以在自定义数据集上实现。 https://gist.github.com/omoindrot/dedc857cdc0e680dfb1be99762990c9c/