Apollo  v5.5.0
Open source self driving car software
Public Member Functions | Protected Member Functions | Protected Attributes | List of all members
apollo::perception::inference::PaddleNet Class Reference

#include <paddle_net.h>

Inheritance diagram for apollo::perception::inference::PaddleNet:
Inheritance graph
Collaboration diagram for apollo::perception::inference::PaddleNet:
Collaboration graph

Public Member Functions

 PaddleNet (const std::string &model_file, const std::string &param_file, const std::vector< std::string > &outputs)
 
 PaddleNet (const std::string &model_file, const std::string &param_file, const std::vector< std::string > &outputs, const std::vector< std::string > &inputs)
 
virtual ~PaddleNet ()
 
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< paddle::PaddlePredictor > predictor_ = nullptr
 
- Protected Attributes inherited from apollo::perception::inference::Inference
int max_batch_size_ = 1
 
int gpu_id_ = 0
 

Constructor & Destructor Documentation

◆ PaddleNet() [1/2]

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

◆ PaddleNet() [2/2]

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

◆ ~PaddleNet()

virtual apollo::perception::inference::PaddleNet::~PaddleNet ( )
inlinevirtual

Member Function Documentation

◆ get_blob()

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

◆ Infer()

void apollo::perception::inference::PaddleNet::Infer ( )
overridevirtual

◆ Init()

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

◆ reshape()

bool apollo::perception::inference::PaddleNet::reshape ( )
protected

◆ shape()

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

Member Data Documentation

◆ predictor_

std::shared_ptr<paddle::PaddlePredictor> apollo::perception::inference::PaddleNet::predictor_ = nullptr
protected

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