Benutzerdefinierte KI-Modelle trainieren

Erster Blick

Sie benötigen einen Computer mit Nvidia GPU, sollten CUDA, CUDNN installieren und das Module Downloader Window verwenden.

Das AugeLab Studio Detection Training-Fenster bietet eine benutzerfreundliche Oberfläche zum Trainieren von Objekterkennungsmodellen mit dem YOLO (You Only Look Once)-Framework.

**Erste Schritte**

1. Starten Sie AugeLab Studio auf Ihrem Computer.

2. Klicken Sie auf AI Tools ➡️ Object Detection Training Widow

3. Bevor Sie mit dem Training beginnen, müssen Sie den Datensatzordner und die Klassendatei laden, die die Klassen der Objekte enthält, die Sie erkennen möchten.

Sie können auch vortrainierte YOLO-Gewichte mit Klassennamen integrieren.

**Menüleiste - Trainingskonfiguration**

1. Datensatz laden: Klicken Sie auf diese Aktion, um den Datensatzordner auszuwählen, der die Bilder für das Training enthält.

2. Klassendatei laden: Verwenden Sie diese Aktion, um die Klassendatei (.names-Format) zu laden, die die Liste der Klassen für das Training enthält.

3. Typ des Erkennungsmodells: Wählen Sie zwischen den Modellen "Schnelle Erkennung" und "Robuste Erkennung". Das Modell "Schnelle Erkennung" eignet sich für PCs mit niedrigen Spezifikationen und bietet eine geringere Genauigkeit. Das Modell "Robuste Erkennung" bietet eine höhere Genauigkeit, erfordert jedoch PCs mit hohen Spezifikationen.

4. Benutzerdefinierte Trainingsprüfung laden: Wenn Sie eine vortrainierte Gewichtedatei (.weights oder .pw-Format) haben, können Sie diese als Ausgangspunkt für das Training laden.

5. Erweiterte Einstellungen öffnen: Weitere (erweiterte) Konfigurationen können im Fenster für erweiterte Einstellungen festgelegt werden.

Fenster für erweiterte Einstellungen

In diesem Fenster können die Trainingsparameter für die erweiterte Verwendung angepasst werden.

1. Datensatzaufteilung: Bestimmt den Anteil der Daten, der für das Training und die Validierung zugewiesen wird, um die Leistung des Modells zu bewerten.

2. Netzwerkgröße: Gibt die Netzwerkgröße an, die den Kompromiss zwischen Geschwindigkeit und Genauigkeit bei Aufgaben der Objekterkennung beeinflusst.

3. Unterteilungsgröße: Definiert die Größe der Mini-Batches während des Trainings, was die Speichernutzung und die Recheneffizienz beeinflusst.

Die Aktion "Train Procedure" im Menü "Hilfe" bietet detaillierte Anweisungen zum effektiven Trainieren der Modelle.

**Starten des Trainingsprozesses**

1. Training starten: Nachdem Sie den Datensatz und die Klassendatei geladen haben, klicken Sie auf START TRAIN, um den Trainingsprozess zu starten.

2. Trainingsprotokoll: Der Protokollbereich zeigt den Fortschritt und den Status des Trainingsprozesses an.

3. Training stoppen: Sie können den Trainingsprozess stoppen, indem Sie auf die Schaltfläche STOP TRAIN klicken.

Sie können den Trainingsprozess von zwei Bildschirmen aus verfolgen:

1. Trainingsprotokollfenster

Training Protokoll

2. Trainingsdiagramm (Verlust und mAP)

Wenn Sie das Fenster zur Objekterkennung schließen, während der Trainingsprozess läuft, wird der Prozess beendet.

Sie können jetzt das Fenster zur Objekterkennung in AugeLab Studio verwenden, um Objekterkennungsmodelle auf benutzerdefinierten Datensätzen zu trainieren. Experimentieren Sie mit verschiedenen Modelltypen, Batchgrößen und benutzerdefinierten Prüfungen, um optimale Ergebnisse zu erzielen! Viel Spaß beim Trainieren Ihrer Objekterkennungsmodelle mit AugeLab Studio!