SampleSSD Class Reference

The SampleSSD class implements the SSD sample. More...

Collaboration diagram for SampleSSD:

Public Member Functions
	SampleSSD (const SampleSSDParams &params)
bool	build ()
	Function builds the network engine. More...
bool	infer ()
	Runs the TensorRT inference engine for this sample. More...
bool	teardown ()
	Cleans up any state created in the sample class. More...

Private Types
template<typename T >
using	SampleUniquePtr = std::unique_ptr< T, samplesCommon::InferDeleter >

Private Member Functions
bool	constructNetwork (SampleUniquePtr< nvinfer1::IBuilder > &builder, SampleUniquePtr< nvinfer1::INetworkDefinition > &network, SampleUniquePtr< nvinfer1::IBuilderConfig > &config, SampleUniquePtr< nvcaffeparser1::ICaffeParser > &parser)
	Parses a Caffe model for SSD and creates a TensorRT network. More...
bool	processInput (const samplesCommon::BufferManager &buffers)
	Reads the input and mean data, preprocesses, and stores the result in a managed buffer. More...
bool	verifyOutput (const samplesCommon::BufferManager &buffers)
	Filters output detections and verify results. More...

Private Attributes
SampleSSDParams	mParams
	The parameters for the sample. More...
nvinfer1::Dims	mInputDims
	The dimensions of the input to the network. More...
std::vector< samplesCommon::PPM< 3, 300, 300 > >	mPPMs
	PPMs of test images. More...
std::shared_ptr< nvinfer1::ICudaEngine >	mEngine
	The TensorRT engine used to run the network. More...

Detailed Description

The SampleSSD class implements the SSD sample.

It creates the network using a caffe model

Member Typedef Documentation

SampleUniquePtr

template<typename T >

using SampleSSD::SampleUniquePtr = std::unique_ptr<T, samplesCommon::InferDeleter>

private

Constructor & Destructor Documentation

SampleSSD()

SampleSSD::SampleSSD ( const SampleSSDParams &  params )

inline

Member Function Documentation

build()

bool SampleSSD::build

(

)

Function builds the network engine.

Creates the network, configures the builder and creates the network engine.

This function creates the SSD network by parsing the caffe model and builds the engine that will be used to run SSD (mEngine)

Returns

Returns true if the engine was created successfully and false otherwise

Here is the call graph for this function:

infer()

bool SampleSSD::infer

(

)

Runs the TensorRT inference engine for this sample.

This function is the main execution function of the sample. It allocates the buffer, sets inputs and executes the engine.

Here is the call graph for this function:

teardown()

bool SampleSSD::teardown

(

)

Cleans up any state created in the sample class.

Clean up the libprotobuf files as the parsing is complete

Note

It is not safe to use any other part of the protocol buffers library after ShutdownProtobufLibrary() has been called.

Here is the call graph for this function:

constructNetwork()

bool SampleSSD::constructNetwork ( SampleUniquePtr< nvinfer1::IBuilder > &  builder, SampleUniquePtr< nvinfer1::INetworkDefinition > &  network, SampleUniquePtr< nvinfer1::IBuilderConfig > &  config, SampleUniquePtr< nvcaffeparser1::ICaffeParser > &  parser  )

private

Parses a Caffe model for SSD and creates a TensorRT network.

Uses a caffe parser to create the SSD Network and marks the output layers.

Parameters

network	Pointer to the network that will be populated with the SSD network
builder	Pointer to the engine builder

Here is the call graph for this function:

Here is the caller graph for this function:

processInput()

bool SampleSSD::processInput ( const samplesCommon::BufferManager &  buffers )

private

Reads the input and mean data, preprocesses, and stores the result in a managed buffer.

Here is the call graph for this function:

Here is the caller graph for this function:

verifyOutput()

bool SampleSSD::verifyOutput ( const samplesCommon::BufferManager &  buffers )

private

Filters output detections and verify results.

Filters output detections and verify result.

Returns

whether the detection output matches expectations

Here is the call graph for this function:

Here is the caller graph for this function:

Member Data Documentation

mParams

SampleSSDParams SampleSSD::mParams

private

The parameters for the sample.

mInputDims

nvinfer1::Dims SampleSSD::mInputDims

private

The dimensions of the input to the network.

mPPMs

std::vector<samplesCommon::PPM<3, 300, 300> > SampleSSD::mPPMs

private

PPMs of test images.

mEngine

std::shared_ptr<nvinfer1::ICudaEngine> SampleSSD::mEngine

private

The TensorRT engine used to run the network.

---

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

• sampleSSD.cpp