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

Vitis AI Library 用户指南 (UG1354)

Document ID
UG1354
Release Date
2022-06-15
Version
2.5 简体中文
此基本类对应 RGBDsegmentation

输入是一对图像,由 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

图 1. 输出图像

函数快速参考

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

表 1. 函数快速参考
类型 成员 实参
std::unique_ptr< RGBDsegmentation > create
  • const std::string & model_name
  • bool need_preprocess
SegmentationResult run
  • const cv::Mat & image_bgr
  • const cv::Mat & image_hha