Trainieren Sie eine Mask R-CNN... Notiz

Trainieren Sie eine Mask R-CNN für die Instanzsegmentierung mit TF Model Garden

Dieses praktische Tutorial zeigt, wie man ein Mask R-CNN mit einem MobileNetV2-Backbone mithilfe von TensorFlow Model Garden feinabstimmt. Sie laden LVIS-Annotationen herunter, generieren TFRecords aus COCO-Bildern, laden eine registrierte Mask R-CNN-Experimentkonfiguration, passen sie für einen kleinen benutzerdefinierten Klassensatz an, legen Trainingshyperparameter fest, führen verteiltes Training durch, evaluieren und exportieren schließlich ein SavedModel für die Bereitstellung. Code und Konfigurationen folgen den Best Practices von TensorFlow; Originalbeispiele sind CC BY 4.0 / Apache 2.0.