apollo::perception::inference::Inference Class Reference

#include <inference.h>

Inheritance diagram for apollo::perception::inference::Inference:

Collaboration diagram for apollo::perception::inference::Inference:

Public Member Functions
virtual void	Infer ()=0
	Inference ()=default
virtual	~Inference ()=default
virtual bool	Init (const std::map< std::string, std::vector< int >> &shapes)=0
void	set_max_batch_size (const int &batch_size)
void	set_gpu_id (const int &gpu_id)
virtual std::shared_ptr< apollo::perception::base::Blob< float > >	get_blob (const std::string &name)=0

Protected Attributes
int	max_batch_size_ = 1
int	gpu_id_ = 0

Constructor & Destructor Documentation

Inference()

apollo::perception::inference::Inference::Inference ( )

default

~Inference()

virtual apollo::perception::inference::Inference::~Inference ( )

virtualdefault

Member Function Documentation

get_blob()

virtual std::shared_ptr<apollo::perception::base::Blob<float> > apollo::perception::inference::Inference::get_blob ( const std::string &  name )

pure virtual

Implemented in apollo::perception::inference::OnnxObstacleDetector, apollo::perception::inference::RTNet, apollo::perception::inference::CaffeNet, apollo::perception::inference::TorchDet, apollo::perception::inference::TorchNet, and apollo::perception::inference::ObstacleDetector.

Infer()

virtual void apollo::perception::inference::Inference::Infer ( )

pure virtual

Implemented in apollo::perception::inference::OnnxObstacleDetector, apollo::perception::inference::RTNet, apollo::perception::inference::CaffeNet, apollo::perception::inference::TorchDet, apollo::perception::inference::TorchNet, and apollo::perception::inference::ObstacleDetector.

Init()

virtual bool apollo::perception::inference::Inference::Init ( const std::map< std::string, std::vector< int >> &  shapes )

pure virtual

Implemented in apollo::perception::inference::OnnxObstacleDetector, apollo::perception::inference::RTNet, apollo::perception::inference::CaffeNet, apollo::perception::inference::TorchDet, apollo::perception::inference::TorchNet, and apollo::perception::inference::ObstacleDetector.

set_gpu_id()

void apollo::perception::inference::Inference::set_gpu_id

(

const int &

gpu_id

)

set_max_batch_size()

void apollo::perception::inference::Inference::set_max_batch_size

(

const int &

batch_size

)

Member Data Documentation

gpu_id_

int apollo::perception::inference::Inference::gpu_id_ = 0

protected

max_batch_size_

int apollo::perception::inference::Inference::max_batch_size_ = 1

protected

---

The documentation for this class was generated from the following file:

• modules/perception/inference/inference.h