[求助!] 使用google object detectionAPI训练自己的数据集产生问题

qq_33443949

于 2018-05-05 21:16:32 发布

阅读量526

点赞数

文章标签： object detection API

本文链接：https://blog.csdn.net/qq_33443949/article/details/80209291

版权

我参考https://blog.csdn.net/dy_guox/article/details/79111949这个博主的教程

使用ssd_mobilenet_v1_coco.config

任务是做“实时建筑物识别”，即将一张图片里的建筑物框出来。

我标注了400张训练集建筑物的图片，60张测试集。图片都类似于下面

然后按照教程开始训练，但是，训练一开始时，就是这样的

然后训练了四万多步LOSS还是乱的

主要训练参数设置如下

train_config: {
  batch_size: 2
  optimizer {
    rms_prop_optimizer: {
      learning_rate: {
        exponential_decay_learning_rate {
          initial_learning_rate: 0.0005
          decay_steps: 800720
          decay_factor: 0.95
        }
      }
      momentum_optimizer_value: 0.9
      decay: 0.9
      epsilon: 1.0
    }
  }

不知道是哪里出了问题