我正在训练Mobilnet-SSD以检测小头。
因此，我想在网络层的前面添加一些box_predictor_layer，例如添加一个box_predictor_layer来连接第5/6卷积层。
该怎么做？
我阅读了所有.proto文件，找到了ssd.proto和ssd_anchor_generator.proto，box_predictor.proto。
但是我不知道如何更改它们，因为张量板的图形是如此无序。
答案 0 :(得分：0)
您需要从主干网中添加另一个“出口点”（即要应用检测的特征图）。为此，请在here中添加图层名称，然后在-1的相应位置添加layer_depth（这意味着您无需告诉它要素的深度是多少）地图，因为它是由骨干给出的。
如果您只需要检测较小的磁头，那么就不需要检测较大的磁头，则可以摆脱（至少某些）附加SSD功能图，这些附加功能图由{ {1}中的{1}}和512、256、256、128。
不要忘记在配置文件中相应地修改from_layer：
layer_depth