配置文件Tensorflow对象检测API

如何解决配置文件Tensorflow对象检测API

我正在尝试使用ssd_mobilenet_v2_320x320_coco17_tpu-8.config训练对象检测api，但要看损失值。这个模型不能完美训练。有人可以帮助我吗？

解决方法

在遵循 tf 对象检测 api 分步指南后，我遇到了同样的“问题”。也许对你来说有点太晚了，但我希望为别人节省时间：你只需要调整学习率。我读过的教程是“tf object detection api training”的前 10 个谷歌搜索结果之一，他们没有调整对于某些模型来说太高的学习率（如果你只是复制了那里提到的管道文件）

请查看您的 config file (line 162)

      learning_rate: {
        cosine_decay_learning_rate {
          learning_rate_base: .8
          total_steps: 20000
          warmup_learning_rate: 0.13333
          warmup_steps: 2000
        }

warmup_steps 意味着您通常以较低的学习率训练模型的前 2000 步（在本例中），以慢慢调整预训练的权重以适应新的挑战。我在 warmup_learning_rate: 0.0005 和 learning_rate_base: 0.008 上看到了一些不错的结果，这显然取决于您的数据和训练的总步骤，但 0.1333 和 0.8 太高了。