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

Question

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

有人可以建议使用最佳方法来微调模型以适应这些图像吗？

或许我可以指点一步一步指导如何做到这一点？道歉，因为我是新手，我正在试图弄清楚如何开始微调。

提前致谢。

Answer 1

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

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