apollo::perception::inference::BatchStream Class Reference

#include <batch_stream.h>

Collaboration diagram for apollo::perception::inference::BatchStream:

Public Member Functions
	BatchStream (int batchSize, int maxBatches, std::string data_path)
	BatchStream ()
void	reset (int firstBatch)
bool	next ()
void	skip (int skipCount)
float *	getBatch ()
int	getBatchesRead () const
int	getBatchSize () const
nvinfer1::DimsNCHW	getDims () const

Constructor & Destructor Documentation

BatchStream() [1/2]

apollo::perception::inference::BatchStream::BatchStream	(	int	batchSize,
		int	maxBatches,
		std::string	data_path
	)

BatchStream() [2/2]

apollo::perception::inference::BatchStream::BatchStream

(

)

Member Function Documentation

getBatch()

float* apollo::perception::inference::BatchStream::getBatch ( )

inline

getBatchesRead()

int apollo::perception::inference::BatchStream::getBatchesRead ( ) const

inline

getBatchSize()

int apollo::perception::inference::BatchStream::getBatchSize ( ) const

inline

getDims()

nvinfer1::DimsNCHW apollo::perception::inference::BatchStream::getDims ( ) const

inline

next()

bool apollo::perception::inference::BatchStream::next

(

)

reset()

void apollo::perception::inference::BatchStream::reset

(

int

firstBatch

)

skip()

void apollo::perception::inference::BatchStream::skip

(

int

skipCount

)

---

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

• modules/perception/inference/tensorrt/batch_stream.h