YOLO-NAS详细教程-训练深度学习分割模型-从图像中去除背景

训练外部模型

在本例中，我们将使用 SuperGradients 训练深度学习分割模型，以从
图像中提取人物肖像，即从图像中去除背景。我们将展示 SuperGradients 如何将外部模型、数据集、损失函数和指标无缝集成到训练流程中。

快速安装

对于这个例子，唯一必要的包是超级梯度。安装超级梯度还将安装运行本示例中的代码所需的所有依赖项。

pip install super-gradients

1. 数据集