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

Vitis AI Library 用户指南 (UG1354)

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

输入是点数据和相关参数。

输出是检测结果的结构体,名为 PointPaintingResult。

代码样本:

  ...
  std::string anno_file_name = "./sample_pointpainting.info";
  PointsInfo points_info;  
  std::vector<cv::Mat> images;
  read_inno_file_pointpainting(anno_file_name, points_info, 5, points_info.sweep_infos, 16, images); 
  std::string seg_model = "semanticfpn_nuimage_576_320_pt";
  std::string model_0 = "pointpainting_nuscenes_40000_64_0_pt";
  std::string model_1 = "pointpainting_nuscenes_40000_64_1_pt";
  auto pointpainting = vitis::ai::PointPainting::create(
       seg_model, model_0, model_1);
  auto ret = pointpainting->run(images, points_info);
  ...
please see the test sample for detail.

函数快速参考

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

表 1. 函数快速参考
类型 成员 实参
std::unique_ptr< PointPainting > create
  • const std::string & seg_model_name
  • const std::string & pp_model_name_0
  • const std::string & pp_model_name_1
int getInputWidth
  • void
int getInputHeight
  • void
size_t get_pointpillars_batch
  • void
size_t get_segmentation_batch
  • void
PointPaintingResult run
std::vector< PointPaintingResult > run
std::vector< cv::Mat > runSegmentation
  • std::vector< cv::Mat > batch_images
std::vector< float > fusion
vitis::ai::pointpillars_nus::PointsInfo runSegmentationFusion
PointPaintingResult runPointPillars
std::vector< PointPaintingResult > runPointPillars