vitis::ai::TFSSDPostProcess

Class of the TFSSD post-process. It initializes the parameters once instead of computing them each time the program executes.

Quick Function Reference

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

Table 1. Quick Function Reference
TypeNameArguments
std::unique_ptr< TFSSDPostProcess >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
TFSSDResultssd_post_process
  • void
std::vector< TFSSDResult >ssd_post_process
  • void

create

Create an TFSSDPostProcess object.

Prototype

std::unique_ptr< TFSSDPostProcess > 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, const std::string &dirname, 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 TFSSDPostProcess.

ssd_post_process

The post-processing function of the TFSSD network.

Prototype


            TFSSDResult ssd_post_process(unsigned int idx)=0;

Returns

The struct of TFSSDResult.

ssd_post_process

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

Prototype

std::vector< TFSSDResult > ssd_post_process()=0;

Returns

The vector of struct of TFSSDResult.