Detailliertes YOLO-NAS-Tutorial – Training des Deep-Learning-Segmentierungsmodells – Entfernen des Hintergrunds aus Bildern

Trainieren Sie ein externes Modell

In diesem Beispiel werden wir SuperGradients verwenden, um ein Deep-Learning-Segmentierungsmodell zu trainieren, um
Porträts aus Bildern zu extrahieren, d. h. den Hintergrund aus dem Bild zu entfernen. Wir zeigen, wie SuperGradients externe Modelle, Datensätze, Verlustfunktionen und Metriken nahtlos in den Trainingsprozess integrieren kann.

Schnelle Installation

Für dieses Beispiel ist SuperGradient das einzige erforderliche Paket. Durch die Installation von Super Gradient werden auch alle Abhängigkeiten installiert, die zum Ausführen des Codes in diesem Beispiel erforderlich sind.

pip install super-gradients

1. Datensatz