用於兩級檢測的更改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。