如何在 Tensorflow Object Detection API 中更改 Mobilnet-SSD 的模型结构

Question

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

怎么办？
我阅读了所有 .proto 文件，找到了 ssd.proto，以及 ssd_anchor_generator.proto，box_predictor.proto。
但我不知道如何改变它们，因为张量板的图太无序了。

Answer 1

您需要从主干添加另一个"出口点"（即要应用检测的特征图）。为此，请在此处添加图层的名称，并在layer_depth中的相应位置-1（这意味着您无需告诉它特征图的深度是多少，因为它是由主干给出的）。如果您只需要检测小磁头，因此不需要检测大磁头，则可能可以摆脱（至少部分）额外的SSD功能图，这些功能图由from_layer中的's和layer_depth中的512,256,256,128添加到骨干之后。不要忘记相应地修改配置文件中的num_layers：

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