vitis::ai::BEVdet - 3.5 日本語

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

Document ID
UG1354
Release Date
2023-06-29
Version
3.5 日本語
入力イメージ (cv::Mat) 内の物体を検出するためのベース クラス。入力はイメージ (cv::Mat) です。出力は入力イメージ内の物体の位置です。サンプル コード:
auto model = vitis::ai::BEVdet::create(argv[1], argv[2], argv[3]);
std::vector<std::string> names;
LoadImageNames(argv[4], names);
std::vector<cv::Mat> images;
for (auto&& i : names) {
  images.emplace_back(cv::imread(i));
}
std::vector<std::vector<char>> bins;
std::vector<std::string> bin_names;
LoadImageNames(argv[5], bin_names);
for (auto&& i : bin_names) {
  auto infile = std::ifstream(i, std::ios_base::binary);
  bins.emplace_back(std::vector<char>(std::istreambuf_iterator<char>(infile),
                                      std::istreambuf_iterator<char>()));
}
std::vector<vitis::ai::CenterPointResult> res;

res = model->run(images, bins);
for (size_t i = 0; i < 32 && i < res.size(); i++) {
  const auto& r = res[i];
  cout << "label: " << r.label << " score: " << r.score
       << " bbox: " << r.bbox[0] << " " << r.bbox[1] << " " << r.bbox[2]
       << " " << r.bbox[3] << " " << r.bbox[4] << " " << r.bbox[5] << " "
       << r.bbox[6] << " " << r.bbox[7] << " " << r.bbox[8] << endl;
}

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

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

表 1. 関数クイック リファレンス
タイプ メンバー 引数
std::unique_ptr< BEVdet > create
  • const std::string & model_name0
  • const std::string & model_name1
  • const std::string & model_name2
std::vector< CenterPointResult > run
  • const std::vector< cv::Mat > & images
  • const std::vector< std::vector< char > > & input_bins