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

Vitis AI Library 用户指南 (UG1354)

Document ID
UG1354
Release Date
2023-06-29
Version
3.5 简体中文
Segmentation8UC3 的类,该类运行函数 run(const cv::Mat& image) 会返回类型为 cv_8UC3 的 cv::Mat。代码样本:
   auto det =
vitis::ai::Segmentation8UC3::create(vitis::ai::SEGMENTATION_FPN);
  auto img = cv::imread("sample_segmentation.jpg");

   int width = det->getInputWidth();
   int height = det->getInputHeight();
   cv::Mat image;
   cv::resize(img, image, cv::Size(width, height), 0, 0,
              cv::INTER_LINEAR);
   auto result = det->run(image);
   cv::imwrite("segres.jpg",result.segmentation);

函数快速参考

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

表 1. 函数快速参考
类型 成员 实参
std::unique_ptr< Segmentation8UC3 > create
  • const std::string & model_name
  • bool need_preprocess
int getInputWidth
  • void
int getInputHeight
  • void
size_t get_input_batch
  • void
SegmentationResult run
  • const cv::Mat & image
std::vector< SegmentationResult > run
  • const std::vector< cv::Mat > & images