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

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

Document ID
UG1354
Release Date
2022-06-15
Version
2.5 日本語
multitask ネットワーク タイプ。multitask ネットワークの宣言。

クラスの番号。ラベル: 0 の名前: "background" ラベル: 1 の名前: "person" ラベル: 2 の名前: "car" ラベル: 3 の名前: "truck" ラベル: 4 の名前: "bus" ラベル: 5 の名前: "bike" ラベル: 6 の名前: "sign" ラベル: 7 の名前: "light"。

イメージ (cv::Mat) からの ADAS MuiltTask のベース クラス。

入力はイメージ (cv::Mat) です。

出力は構造体 MultiTaskResult で、セグメンテーション結果、検出結果、vehicle towards を含みます。

サンプル コード:

auto det = vitis::ai::MultiTask::create("multi_task");
auto image = cv::imread("sample_multitask.jpg");
auto result = det->run_8UC3(image);
cv::imwrite("sample_multitask_result.jpg",result.segmentation);

モデルの結果を次に示します。

図 1. 結果イメージ

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

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

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