用于两级检测的更改SSD网络（Single Shot Multibox Detector）

Question

我已经使用此链接：https://github.com/albanie/wider2pascal将Widerface数据集注释更改为Pascalvoc格式，因为SSD网络最初是为PascalVoc数据集编写的。 现在我们要运行SSD网络。我们应该如何将SSD更改为两个类别检测？代码很复杂，我们很困惑哪些行应该改变。

请帮助我们。

Answer 1

This SSD implementation提供了关于如何使用受过训练的SSD在您自己的数据集上进行传输学习的工具和教程。该解决方案（或至少一个可能的解决方案，本教程解释了多个备选方案）是对经过训练的SSD的分类器图层的权重张量进行子采样，以使它们的形状与数据集中类的数量相符。

Here is the tutorial。