vitis::ai::Unet2D - 3.5 简体中文 - UG1354

Vitis AI Library 用户指南 (UG1354)

Document ID
UG1354
Release Date
2023-06-29
Version
3.5 简体中文
此基本类对应 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 类中定义的所有函数:

表 1. 函数快速参考
类型 成员 实参
std::unique_ptr< Unet2D > create
  • const std::string & model_name
  • bool need_preprocess
vitis::ai::Unet2DResult run
  • float * img
  • int len
vitis::ai::Unet2DResult run
  • const std::vector< float > & img
std::vector< vitis::ai::Unet2DResult > run
  • const std::vector< float * > & imgs
  • int len
std::vector< vitis::ai::Unet2DResult > run
  • const std::vector< std::vector< float > > & imgs
int getInputWidth
  • void
int getInputHeight
  • void
size_t get_input_batch
  • void