2クラス検出のためのSSDネットワーク（シングルショットマルチボックス検出器）の変更

Question

このリンクを使用して：https://github.com/albanie/wider2pascal私はSSDネットワークがもともとPascalVocデータセット用に書かれているため、Widerfaceデータセット注釈をPascalvocフォーマットに変更しました。 これでSSDネットワークを実行します。 SSDを2つのクラス検出に変更するにはどうすればよいですか？コードは複雑で、混乱しています。どの行を変更する必要がありますか？

ご協力ください。

Answer 1

This SSD implementationには、訓練されたSSDを使用して自分のデータセットで転送学習を行う方法に関するツールとチュートリアルが用意されています。解決策（または少なくとも1つの可能な解決策、チュートリアルでは複数の選択肢について説明しています）は、訓練されたSSDの分類子層のウェイトテンソルをサブサンプリングして、その形状がデータセットのクラス数に適合するようにします。

Here is the tutorial。