更改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