vitis::ai::Unet2D - 3.5 English - UG1354

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: