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

训练外部模型

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

快速安装

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

pip install super-gradients

1. 数据集

你可能感兴趣的:(深度学习开发实践系列,YOLO,深度学习,人工智能,计算机视觉,机器学习)