vitis::ai::MedicalSegmentationPostProcess

Class of the MedicalSegmentation post-process. It will initialize the parameters once instead of computing them every time when the program executes.

Quick Function Reference

The following table lists all the functions defined in the vitis::ai::MedicalSegmentationPostProcess class:

Table 1. Quick Function Reference
TypeNameArguments
std::unique_ptr< MedicalSegmentationPostProcess >create
  • const std::vector< vitis::ai::library::InputTensor > & input_tensors
  • const std::vector< vitis::ai::library::OutputTensor > & output_tensors
  • const vitis::ai::proto::DpuModelParam & config
MedicalSegmentationResultmedicalsegmentation_post_process
  • void
std::vector< MedicalSegmentationResult >medicalsegmentation_post_process
  • void

create

Create an MedicalSegmentationPostProcess object.

Prototype

std::unique_ptr< MedicalSegmentationPostProcess > create(const std::vector< vitis::ai::library::InputTensor > &input_tensors, const std::vector< vitis::ai::library::OutputTensor > &output_tensors, const vitis::ai::proto::DpuModelParam &config, int &real_batch_size);

Parameters

The following table lists the create function arguments.

Table 2. create Arguments
Type Name Description
const std::vector< vitis::ai::library::InputTensor > & input_tensors A vector of all input-tensors in the network. Usage: input_tensors[input_tensor_index].
const std::vector< vitis::ai::library::OutputTensor > & output_tensors A vector of all output-tensors in the network. Usage: output_tensors[output_index].
const vitis::ai::proto::DpuModelParam & config The DPU model configuration information.

Returns

A unique printer of MedicalSegmentationPostProcess.

medicalsegmentation_post_process

The post-processing function of the MedicalSegmentation network.

Prototype


            MedicalSegmentationResult medicalsegmentation_post_process(unsigned int idx)=0;

Returns

The struct of MedicalSegmentationResult.

medicalsegmentation_post_process

The batch mode post-processing function of the MedicalSegmentation network.

Prototype

std::vector< MedicalSegmentationResult > medicalsegmentation_post_process()=0;

Returns

The vector of struct of MedicalSegmentationResult.