如何在 Tensorflow 对象检测 API 中更改 Mobilnet-SSD 的模型结构？

Question

我正在训练一个 Mobilnet-SSD 来检测小头。
所以我想在网络层的前面添加一些box_predictor_layer，比如添加一个box_predictor_layer来连接5/6th Convolution层。

怎么做？
我阅读了所有 .proto 文件，找到了 ssd.proto、ssd_anchor_generator.proto、box_predictor.proto。
但是不知道怎么改，因为tensorboard的图太乱了。

Answer 1

您需要从主干添加另一个“退出点”（即应用检测的特征图）。通过在here 中添加层的名称，并在layer_depth 的相应位置添加-1 来做到这一点（这意味着您不需要告诉它特征图的深度是多少，因为它由下式给出骨干）。 如果您只需要检测小头，而不需要检测大头，您可能可以摆脱（至少部分）附加的 SSD 特征图，这些特征图由 @ 中的 ''s 添加到主干之后987654325@ 和 512、256、256、128 在layer_depth。 不要忘记相应地修改配置文件中的num_layers：

model {
  ssd {
    anchor_generator {
      ssd_anchor_generator {
        num_layers: <num_of_feature_maps>
        ...
      }
    }
  }
}