SampleMaskRCNN Class Reference

Collaboration diagram for SampleMaskRCNN:

Public Member Functions
	SampleMaskRCNN (const SampleMaskRCNNParams &params)
bool	build ()
bool	infer ()
bool	teardown ()

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< nvuffparser::IUffParser > &parser)
bool	processInput (const samplesCommon::BufferManager &buffers)
bool	verifyOutput (const samplesCommon::BufferManager &buffers)
std::vector< MaskRCNNUtils::BBoxInfo >	decodeOutput (const int imageIdx, void *detectionsHost, void *masksHost)

Private Attributes
SampleMaskRCNNParams	mParams
nvinfer1::Dims	mInputDims
std::vector< MaskRCNNUtils::PPM< uint8_t > >	mOriginalPPMs
std::vector< MaskRCNNUtils::PPM< uint8_t > >	mPPMs
std::shared_ptr< nvinfer1::ICudaEngine >	mEngine

Member Typedef Documentation

SampleUniquePtr

template<typename T >

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

private

Constructor & Destructor Documentation

SampleMaskRCNN()

SampleMaskRCNN::SampleMaskRCNN ( const SampleMaskRCNNParams &  params )

inline

Member Function Documentation

build()

bool SampleMaskRCNN::build

(

)

Here is the call graph for this function:

infer()

bool SampleMaskRCNN::infer

(

)

Here is the call graph for this function:

teardown()

bool SampleMaskRCNN::teardown

(

)

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 SampleMaskRCNN::constructNetwork ( SampleUniquePtr< nvinfer1::IBuilder > &  builder, SampleUniquePtr< nvinfer1::INetworkDefinition > &  network, SampleUniquePtr< nvuffparser::IUffParser > &  parser  )

private

Here is the call graph for this function:

Here is the caller graph for this function:

processInput()

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

private

Here is the call graph for this function:

Here is the caller graph for this function:

verifyOutput()

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

private

Here is the call graph for this function:

Here is the caller graph for this function:

decodeOutput()

std::vector< MaskRCNNUtils::BBoxInfo > SampleMaskRCNN::decodeOutput ( const int  imageIdx, void *  detectionsHost, void *  masksHost  )

private

Here is the call graph for this function:

Here is the caller graph for this function:

Member Data Documentation

mParams

SampleMaskRCNNParams SampleMaskRCNN::mParams

private

mInputDims

nvinfer1::Dims SampleMaskRCNN::mInputDims

private

mOriginalPPMs

std::vector<MaskRCNNUtils::PPM<uint8_t> > SampleMaskRCNN::mOriginalPPMs

private

mPPMs

std::vector<MaskRCNNUtils::PPM<uint8_t> > SampleMaskRCNN::mPPMs

private

mEngine

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

private

---

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

• sampleUffMaskRCNN.cpp