vitis::ai::MultiTaskPostProcess

Quick Function Reference

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

Table 1. Quick Function Reference
TypeNameArguments
std::unique_ptr< MultiTaskPostProcess >create
  • const std::vector< std::vector< vitis::ai::library::InputTensor >> & input_tensors
  • const std::vector< std::vector< vitis::ai::library::OutputTensor >> & output_tensors
  • const vitis::ai::proto::DpuModelParam & config
std::vector< MultiTaskResult >post_process_seg
  • void
std::vector< MultiTaskResult >post_process_seg_visualization
  • void

create

Factory function to get an instance of derived classes of MultiTaskPostProcess.

Prototype

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

Parameters

The following table lists the create function arguments.

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

Returns

The struct of MultiTaskResult.

post_process_seg

The post-processing function of the multitask which stored the original segmentation classes.

Prototype

std::vector< MultiTaskResult > post_process_seg(size_t batch_size)=0;

Returns

The struct of SegmentationResult.

post_process_seg_visualization

The post-processing function of the multitask which return a result include segmentation image mapped to color.

Prototype

std::vector< MultiTaskResult > post_process_seg_visualization(size_t batch_size)=0;

Returns

The struct of SegmentationResult.