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

Fine-tuning a pre-trained VGG16 model on my own data using TF-Slim

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

谁能建议最好的方法来微调模型以适应这些图像?

或者可以为我指明如何执行此操作的分步指南的方向?抱歉,我是 convnets 的新手,我正在尝试弄清楚如何开始微调。

提前致谢。

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

此外,我建议您查看以下 git 以在自定义数据集上实施。 https://gist.github.com/omoindrot/dedc857cdc0e680dfb1be99762990c9c/