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

Vitis AI Library 用户指南 (UG1354)

Document ID
UG1354
Release Date
2023-06-29
Version
3.5 简体中文
此基本类用于检测来自图像 (cv::Mat) 的 rcan。

输入为图像 (cv::Mat)。

输出为放大的图像。

代码样本:

注释: 输入图像大小为 640x360
if (argc < 2) {
  cerr << "usage: " << argv[0] << "  modelname  image_file_url " << endl;
  abort();
}
Mat input_img = imread(argv[2]);
if (input_img.empty()) {
  cerr << "can't load image! " << argv[2] << endl;
  return -1;
}
auto det = vitis::ai::Rcan::create(argv[1]);
Mat ret_img = det->run(input_img).feat;
imwrite("sample_rcan_result.png", ret_img);

显示模型结果:

图 1. 结果图像
result image

函数快速参考

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

表 1. 函数快速参考
类型 成员 实参
std::unique_ptr< Rcan > create
  • const std::string & model_name
  • bool need_preprocess
RcanResult run
  • const cv::Mat & image
std::vector< RcanResult > run
  • const std::vector< cv::Mat > & images
std::vector< RcanResult > run
  • const std::vector< vart::xrt_bo_t > & input_bos