vitis::ai::CenterPoint

Quick Function Reference

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

Table 1. Quick Function Reference
TypeNameArguments
std::unique_ptr< CenterPoint >create
  • void
intgetInputWidth
  • void
intgetInputHeight
  • void
size_tget_input_batch
  • void
std::vector< CenterPointResult >run
  • void
std::vector< std::vector< CenterPointResult > >run
  • void

create

Factory function to get an instance of derived classes of class CenterPoint.

value is true.

Prototype

std::unique_ptr< CenterPoint > create(const std::string &model_name_0, const std::string &model_name_1);

Returns

An instance of CenterPoint class.

getInputWidth

Function to get InputWidth of the centerpoint network (input image columns).

Prototype

int getInputWidth() const =0;

Returns

InputWidth of the centerpoint network

getInputHeight

Function to get InputHeight of the centerpoint network (input image rows).

Prototype

int getInputHeight() const =0;

Returns

InputHeight of the centerpoint network.

get_input_batch

Function to get the number of images processed by the DPU at one time.

Note: Different DPU core the batch size may be different. This depends on the IP used.

Prototype

size_t get_input_batch() const =0;

Returns

Batch size.

run

Prototype

std::vector< CenterPointResult > run(const std::vector< float > &input)=0;

run

Prototype

std::vector< std::vector< CenterPointResult > > run(const std::vector< std::vector< float >> &inputs)=0;