Retrain Objekterkennung Modell mit eigenen Bildern (tensorflow)

Question

Guten Morgen,

Ich habe mit dem tensorflow Arbeits object detection tutorial mit dem ssd_mobilenet sie als gefrorene Diagramm sowie mit den entsprechenden Checkpoint-Dateien (Modell bieten. ckpt.data-00000-of-00001, model.ckpt.index, model.ckpt.meta). Da die Bilder jedoch manchmal schlecht erkannt werden, hoffte ich, dass ich dem Erkennungsmodell eigene Bilder zuführen und seine Leistung für meine Bilder verbessern konnte, die alle von derselben Kamera aufgenommen wurden.

Google konnte mir nicht helfen, wo ich anfangen soll. Die Fragen, die ich habe: - Gibt es Code-Schnipsel, die zeigen, welche dieser Dateien zu laden und wie das vorhandene Modell zu trainieren? - Muss ich das geladene Modell mit den alten Daten (d. H. COCO) + den neuen Daten (meine Bilder) umschulen oder kann ich es einfach mit meinen Daten neu trainieren und das Modell merkt sich, was es vorher gelernt hat?

Entschuldigung für diese sehr unspezifischen Fragen, aber ich kann einfach nicht herausfinden, wo ich anfangen soll.

Mit freundlichen Grüßen,

gbrown

Answer 1

Es gibt eine große walkthrough blog und code base von Dat Tran geschrieben. Er trainierte ein Modell zur Erkennung von Raccoons in Bildern mit dem vortrainierten SSD_mobilenet als Starthilfe. Dies ist der beste Ort, den ich gefunden habe. Hoffe das hilft.