apollo::perception::inference::TorchDet Class Reference

#include <torch_det.h>

Inheritance diagram for apollo::perception::inference::TorchDet:

Collaboration diagram for apollo::perception::inference::TorchDet:

Public Member Functions
	TorchDet (const std::string &net_file, const std::string &model_file, const std::vector< std::string > &outputs)
	TorchDet (const std::string &net_file, const std::string &model_file, const std::vector< std::string > &outputs, const std::vector< std::string > &inputs)
virtual	~TorchDet ()
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	shape (const std::string &name, std::vector< int > *res)

Protected Attributes
torch::jit::script::Module	net_
Protected Attributes inherited from apollo::perception::inference::Inference
int	max_batch_size_ = 1
int	gpu_id_ = 0

Constructor & Destructor Documentation

TorchDet() [1/2]

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

TorchDet() [2/2]

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

~TorchDet()

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

inlinevirtual

Member Function Documentation

get_blob()

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

overridevirtual

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

Infer()

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

overridevirtual

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

Init()

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

overridevirtual

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

shape()

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

protected

Member Data Documentation

net_

torch::jit::script::Module apollo::perception::inference::TorchDet::net_

protected

---

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

• modules/perception/inference/libtorch/torch_det.h