vitis::ai::YOLOv2 - 2.5 日本語

Vitis AI ライブラリ ユーザー ガイド (UG1354)

Document ID
UG1354
Release Date
2022-06-15
Version
2.5 日本語
入力イメージ (cv::Mat) 内の物体を検出するためのベース クラス。入力はイメージ (cv::Mat) です。出力は、入力イメージ内の物体の位置です。サンプル コード:
 auto img = cv::imread("sample_yolov2.jpg");
 auto model = vitis::ai::YOLOv2::create("yolov2_voc");
 auto result = model->run(img);
 for (const auto &bbox : result.bboxes) {
   int label = bbox.label;
   float xmin = bbox.x * img.cols + 1;
   float ymin = bbox.y * img.rows + 1;
   float xmax = xmin + bbox.width * img.cols;
   float ymax = ymin + bbox.height * img.rows;
   if (xmax > img.cols)
     xmax = img.cols;
   if (ymax > img.rows)
     ymax = img.rows;
   float confidence = bbox.score;

   cout << "RESULT: " << label << "\t" << xmin << "\t" << ymin << "\t" << xmax
       << "\t" << ymax << "\t" << confidence << "\n";
   rectangle(img, Point(xmin, ymin), Point(xmax, ymax), Scalar(0, 255, 0), 1,
            1, 0);
}

関数クイック リファレンス

次の表に、vitis::ai::YOLOv2 クラスに定義されているすべての関数を示します。

表 1. 関数クイック リファレンス
タイプ メンバー 引数
std::unique_ptr< YOLOv2 > create
  • const std::string & model_name
  • bool need_preprocess
YOLOv2Result run
  • const cv::Mat & image
std::vector< YOLOv2Result > run
  • const std::vector< cv::Mat > & images
int getInputWidth
  • void
int getInputHeight
  • void
size_t get_input_batch
  • void