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

Vitis AI Library 用户指南 (UG1354)

Document ID
UG1354
Release Date
2022-06-15
Version
2.5 简体中文
此基本类用于评估面部图像 (cv::Mat) 的质量和 5 个关键点的坐标。

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

输出是输入图像中的人脸的质量和 5 个关键点的坐标。

代码样本:显示 FaceQuality5pt 模型结果:width=\textwidth

cv:Mat image = cv::imread("sample_facequality5pt.jpg");
auto network =
      vitis::ai::FaceQuality5pt::create("face-quality", true);
auto result = network->run(image);
auto quality = result.score;
auto points = result.points;
for(int i = 0; i< 5 ; ++i){
    auto x = points[i].frist  * image.cols;
    auto y = points[j].second * image.rows;
}
注释: 默认模式为日间,如果使用日夜切换网络,并且输入图像背景为夜间,请使用 API setMode
network->setMode(vitis::ai::FaceQuality5pt::Mode::NIGHT);
图 1. 结果图像

函数快速参考

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

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