Unet2D
のベース クラス。入力は 4 チャネルのバイナリ データ (NxNx4) です。
出力は Unet2DResult という名前のセグメンテーション結果の構造体です。
サンプル コード:
std::vector<float> vf = get_binary_data();
auto Unet2D = vitis::ai::Unet2D::create("unet2d_tf");
auto result = Unet2D->run(vf.data(), vf.size());
std::cout << result.data.size() << "\n";関数クイック リファレンス
次の表に、vitis::ai::Unet2D クラスに定義されているすべての関数を示します。
| タイプ | メンバー | 引数 |
|---|---|---|
std::unique_ptr<
Unet2D
> |
create |
|
| vitis::ai::Unet2DResult | run |
|
| vitis::ai::Unet2DResult | run |
|
| std::vector< vitis::ai::Unet2DResult > | run |
|
| std::vector< vitis::ai::Unet2DResult > | run |
|
| int | getInputWidth |
|
| int | getInputHeight |
|
| size_t | get_input_batch |
|