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

Question

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

怎么做？
我阅读了所有的.proto文件，找到ssd.proto，和ssd_anchor_generator.proto，box_predictor.proto。
但我不知道怎么改，因为张量板的图形太乱了。在

Answer 1

您需要从主干网添加另一个“出口点”（即应用检测的特征映射）。为此，请在here中添加层的名称，并在layer_depth中的相应位置添加{}（这意味着您不需要告诉它特征映射的深度是多少，因为它是由主干网给出的）。 如果您只需要检测小磁头，所以不需要检测大磁头，那么您可能可以去除（至少部分）附加的SSD特征映射，这些特征映射是由from_layer和layer_depth中的512、256、256、128添加在主干之后。 别忘了相应地修改配置文件中的num_layers：

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