Modifier le réseau SSD (Détecteur multibox monocoup) pour la détection de deux classes

Question

J'ai utilisé ce lien: https://github.com/albanie/wider2pascal pour modifier les annotations de jeu de données Widerface au format Pascalvoc, car le réseau SSD est écrit à l'origine pour l'ensemble de données PascalVoc. Maintenant, nous voulons lancer un réseau SSD. Que devrions-nous faire pour changer SSD à la détection de deux classes? Le code est compliqué et nous sommes confus Quelles lignes devraient être modifiées.

Aidez-nous s'il vous plaît.

Answer 1

This SSD implementation fournit des outils et un tutoriel sur la façon d'utiliser un SSD formé pour l'apprentissage de transfert sur votre propre ensemble de données. La solution (ou au moins une solution possible, le tutoriel explique plusieurs alternatives) consiste à sous-échantillonner les tenseurs de poids des couches de classifieur du SSD entraîné afin que leur forme corresponde au nombre de classes de votre ensemble de données.

Here is the tutorial.