Wolfram 语言

按语义分割图像

语义分割是场景理解的一个高级任务:应用程序包括自动驾驶车辆、虚拟现实、人机交互等。图像分类对整个图像进行预测,语义分割旨在将图像的每个像素分类成组成分类 (constituent class)。下面的例子显示如何使用预先训练好的图像分割网络。还可以用 NetTrain 针对特定数据对模型进行微调。

从 Neural Net Repository 获取预先训练好的网络,并从资源页面复制计算函数。

定义此模型的标签列表。模型输出中的整数对应于标签列表中的元素。

编写一个函数,按像素的分割类可视化像素,并在原始图像上显示分割。虽然不完美,但可以看到大多数对象是被正确分割的。

相关范例

de en es fr ja ko pt-br