apollo::perception::inference::CaffeNet Class Reference

#include <caffe_net.h>

Inheritance diagram for apollo::perception::inference::CaffeNet:

Collaboration diagram for apollo::perception::inference::CaffeNet:

Public Member Functions
	CaffeNet (const std::string &net_file, const std::string &model_file, const std::vector< std::string > &outputs)
	CaffeNet (const std::string &net_file, const std::string &model_file, const std::vector< std::string > &outputs, const std::vector< std::string > &inputs)
virtual	~CaffeNet ()
bool	Init (const std::map< std::string, std::vector< int >> &shapes) override
void	Infer () override
BlobPtr	get_blob (const std::string &name) override
Public Member Functions inherited from apollo::perception::inference::Inference
	Inference ()=default
virtual	~Inference ()=default
void	set_max_batch_size (const int &batch_size)
void	set_gpu_id (const int &gpu_id)

Protected Member Functions
bool	reshape ()
bool	shape (const std::string &name, std::vector< int > *res)

Protected Attributes
std::shared_ptr< caffe::Net< float > >	net_ = nullptr
Protected Attributes inherited from apollo::perception::inference::Inference
int	max_batch_size_ = 1
int	gpu_id_ = 0

Constructor & Destructor Documentation

CaffeNet() [1/2]

apollo::perception::inference::CaffeNet::CaffeNet	(	const std::string &	net_file,
		const std::string &	model_file,
		const std::vector< std::string > &	outputs
	)

CaffeNet() [2/2]

apollo::perception::inference::CaffeNet::CaffeNet	(	const std::string &	net_file,
		const std::string &	model_file,
		const std::vector< std::string > &	outputs,
		const std::vector< std::string > &	inputs
	)

~CaffeNet()

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

inlinevirtual

Member Function Documentation

get_blob()

BlobPtr apollo::perception::inference::CaffeNet::get_blob ( const std::string &  name )

overridevirtual

Implements apollo::perception::inference::Inference.

Infer()

void apollo::perception::inference::CaffeNet::Infer ( )

overridevirtual

Implements apollo::perception::inference::Inference.

Init()

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

overridevirtual

Implements apollo::perception::inference::Inference.

reshape()

bool apollo::perception::inference::CaffeNet::reshape ( )

protected

shape()

bool apollo::perception::inference::CaffeNet::shape ( const std::string &  name, std::vector< int > *  res  )

protected

Member Data Documentation

net_

std::shared_ptr<caffe::Net<float> > apollo::perception::inference::CaffeNet::net_ = nullptr

protected

---

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

• modules/perception/inference/caffe/caffe_net.h