Base class for
Unet2D
. Input is an 4 channel binary data: NxNx4
Output is a struct of segmentation results, named Unet2DResult.
Sample code:
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";
Quick Function Reference
The following table lists all the functions defined in the vitis::ai::Unet2D
class:
Type | Member | Arguments |
---|---|---|
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 |
|