modèle de détection d'objet Reformer avec ses propres images (tensorflow)

Question

Bonjour,

Je travaille avec le tensorflow object detection tutorial en utilisant la ssd_mobilenet qu'ils fournissent sous forme de graphique congelé ainsi qu'avec les fichiers de point de contrôle correspondant (de modèle. ckpt.data-00000-of-00001, model.ckpt.index, model.ckpt.meta). Cependant, comme les images sont parfois mal reconnues, j'espérais pouvoir alimenter mes propres images au modèle de détection et améliorer ses performances pour mes images, toutes prises par le même appareil photo.

Google n'a pas pu m'aider par où commencer. Les questions que je me pose: - Existe-t-il des extraits de code indiquant lesquels de ces fichiers charger et comment former le modèle existant? - Ai-je besoin de recycler le modèle chargé avec les anciennes données (c.-à-d. COCO) + les nouvelles données (mes images) ou puis-je simplement le recycler en utilisant mes données et le modèle se souvient-il de ce qu'il a appris auparavant?

Désolé pour ces questions très non spécifiques, mais je ne peux pas savoir par où commencer.

Meilleures salutations,

gbrown

Answer 1

Il y a une grande walkthrough blog et code base écrit par Dat Tran. Il a formé un modèle pour reconnaître les ratons laveurs dans les images en utilisant le SSD_mobilenet pré-formé comme un début. C'est le meilleur endroit que j'ai trouvé pour commencer. J'espère que cela t'aides.