TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Pages
MaskRCNNUtils Namespace Reference

Classes

struct  BBoxInfo
 
struct  Mask
 
struct  PPM
 
struct  RawDetection
 

Functions

void readPPMFile (const std::string &filename, PPM< uint8_t > &ppm)
 
void writePPMFile (const std::string &filename, PPM< uint8_t > &ppm)
 
template<typename T >
void resizePPM (const PPM< T > &src, PPM< T > &dst, int target_height, int target_width, int channel)
 
void padPPM (const PPM< uint8_t > &src, PPM< uint8_t > &dst, int top, int bottom, int left, int right)
 
void preprocessPPM (PPM< uint8_t > &src, PPM< uint8_t > &dst, int target_h, int target_w)
 
PPM< uint8_t > resizeMask (const BBoxInfo &box, const float mask_threshold)
 
void maskPPM (PPM< uint8_t > &image, const PPM< uint8_t > &mask, const int start_x, const int start_y, const std::vector< int > &color)
 
void addBBoxPPM (PPM< uint8_t > &ppm, const BBoxInfo &box, const PPM< uint8_t > &resized_mask)
 

Function Documentation

◆ readPPMFile()

void MaskRCNNUtils::readPPMFile ( const std::string &  filename,
PPM< uint8_t > &  ppm 
)
Here is the caller graph for this function:

◆ writePPMFile()

void MaskRCNNUtils::writePPMFile ( const std::string &  filename,
PPM< uint8_t > &  ppm 
)
Here is the caller graph for this function:

◆ resizePPM()

template<typename T >
void MaskRCNNUtils::resizePPM ( const PPM< T > &  src,
PPM< T > &  dst,
int  target_height,
int  target_width,
int  channel 
)

◆ padPPM()

void MaskRCNNUtils::padPPM ( const PPM< uint8_t > &  src,
PPM< uint8_t > &  dst,
int  top,
int  bottom,
int  left,
int  right 
)
Here is the caller graph for this function:

◆ preprocessPPM()

void MaskRCNNUtils::preprocessPPM ( PPM< uint8_t > &  src,
PPM< uint8_t > &  dst,
int  target_h,
int  target_w 
)
Here is the call graph for this function:
Here is the caller graph for this function:

◆ resizeMask()

PPM<uint8_t> MaskRCNNUtils::resizeMask ( const BBoxInfo box,
const float  mask_threshold 
)
Here is the caller graph for this function:

◆ maskPPM()

void MaskRCNNUtils::maskPPM ( PPM< uint8_t > &  image,
const PPM< uint8_t > &  mask,
const int  start_x,
const int  start_y,
const std::vector< int > &  color 
)
Here is the caller graph for this function:

◆ addBBoxPPM()

void MaskRCNNUtils::addBBoxPPM ( PPM< uint8_t > &  ppm,
const BBoxInfo box,
const PPM< uint8_t > &  resized_mask 
)
Here is the call graph for this function:
Here is the caller graph for this function: