vitis::ai::RGBDsegmentation - 3.5 简体中文

输入是一对图像，由 RGB 图像 (cv::Mat) 和以深度映射生成的 HHA 映射 (cv::Mat) 组成。

输出是热图，其中每个像素均以一个语义类别（例如，椅子、床和常见室内物体）来预测。

代码样本：

Mat img_bgr = cv::imread("sample_rgbdsegmentation_bgr.jpg");
Mat img_hha = cv::imread("sample_rgbdsegmentation_hha.jpg");

auto segmentation = vitis::ai::RGBDsegmentation::create("SA-Gate_pt", true);

auto result = segmentation->run(img_bgr, img_hha);

imwrite("result.jpg", result.segmentation);

显示模型结果：width=\textwidth

函数快速参考

下表列出了 vitis::ai::RGBDsegmentation 类中定义的所有函数：