此基本类对应
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 |
|