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

Vitis AI Library 用户指南 (UG1354)

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

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

输出图像 reid 特征。

代码样本:

if(argc < 3){
    cerr<<"need two images"<<endl;
    return -1;
}
Mat imgx = imread(argv[1]);
if(imgx.empty()){
    cerr<<"can't load image! "<<argv[1]<<endl;
    return -1;
}
Mat imgy = imread(argv[2]);
if(imgy.empty()){
    cerr<<"can't load image! "<<argv[2]<<endl;
    return -1;
}
auto det = vitis::ai::Reid::create("reid");
Mat featx = det->run(imgx).feat;
Mat featy = det->run(imgy).feat;
double dismat= cosine_distance(featx, featy);
printf("dismat : %.3lf \n", dismat);

函数快速参考

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

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