Namespaces | |
detail | |
Typedefs | |
typedef LReLU | PReLU |
Enumerations | |
enum | CodeTypeSSD : int32_t { CodeTypeSSD::CORNER = 0, CodeTypeSSD::CENTER_SIZE = 1, CodeTypeSSD::CORNER_SIZE = 2, CodeTypeSSD::TF_CENTER = 3 } |
The type of encoding used for decoding the bounding boxes and loc_data. More... | |
Functions | |
__attribute__ ((deprecated)) typedef struct | |
The Region plugin layer performs region proposal calculation: generate 5 bounding boxes per cell (for yolo9000, generate 3 bounding boxes per cell). More... | |
void | throwCudaError (const char *file, const char *function, int line, int status, const char *msg) |
void | throwCublasError (const char *file, const char *function, int line, int status, const char *msg) |
void | throwCudnnError (const char *file, const char *function, int line, int status, const char *msg) |
void | logError (const char *msg, const char *file, const char *fn, int line) |
void | reportAssertion (const char *msg, const char *file, int line) |
size_t | detectionInferenceWorkspaceSize (bool shareLocation, int N, int C1, int C2, int numClasses, int numPredsPerClass, int topK, DataType DT_BBOX, DataType DT_SCORE) |
template<typename T > | |
void | write (char *&buffer, const T &val) |
template<typename T > | |
T | read (const char *&buffer) |
template<typename T > | |
void | scaleShiftChannelsInplace (T *inOut, const int B, const int C, const int channelVolume, const float *beta, const float *gamma, cudaStream_t stream) |
template<typename CreatorType > | |
void | initializePlugin (void *logger, const char *libNamespace) |
typedef LReLU nvinfer1::plugin::PReLU |
|
strong |
struct nvinfer1::plugin::__attribute__ | ( | (deprecated) | ) |
The Region plugin layer performs region proposal calculation: generate 5 bounding boxes per cell (for yolo9000, generate 3 bounding boxes per cell).
For each box, calculating its probablities of objects detections from 80 pre-defined classifications (yolo9000 has 9418 pre-defined classifications, and these 9418 items are organized as work-tree structure). RegionParameters defines a set of parameters for creating the Region plugin layer.
num | Number of predicted bounding box for each grid cell. |
coords | Number of coordinates for a bounding box. |
classes | Number of classfications to be predicted. |
softmaxTree | When performing yolo9000, softmaxTree is helping to do softmax on confidence scores, for element to get the precise classfication through word-tree structured classfication definition. |
void nvinfer1::plugin::throwCudaError | ( | const char * | file, |
const char * | function, | ||
int | line, | ||
int | status, | ||
const char * | msg | ||
) |
void nvinfer1::plugin::throwCublasError | ( | const char * | file, |
const char * | function, | ||
int | line, | ||
int | status, | ||
const char * | msg | ||
) |
void nvinfer1::plugin::throwCudnnError | ( | const char * | file, |
const char * | function, | ||
int | line, | ||
int | status, | ||
const char * | msg | ||
) |
void nvinfer1::plugin::logError | ( | const char * | msg, |
const char * | file, | ||
const char * | fn, | ||
int | line | ||
) |
void nvinfer1::plugin::reportAssertion | ( | const char * | msg, |
const char * | file, | ||
int | line | ||
) |
size_t nvinfer1::plugin::detectionInferenceWorkspaceSize | ( | bool | shareLocation, |
int | N, | ||
int | C1, | ||
int | C2, | ||
int | numClasses, | ||
int | numPredsPerClass, | ||
int | topK, | ||
DataType | DT_BBOX, | ||
DataType | DT_SCORE | ||
) |
void nvinfer1::plugin::write | ( | char *& | buffer, |
const T & | val | ||
) |
T nvinfer1::plugin::read | ( | const char *& | buffer | ) |
void nvinfer1::plugin::scaleShiftChannelsInplace | ( | T * | inOut, |
const int | B, | ||
const int | C, | ||
const int | channelVolume, | ||
const float * | beta, | ||
const float * | gamma, | ||
cudaStream_t | stream | ||
) |
void nvinfer1::plugin::initializePlugin | ( | void * | logger, |
const char * | libNamespace | ||
) |
nvinfer1::plugin::softmaxTree |
ILogger * nvinfer1::plugin::gLogger {} |
LogStream< ILogger::Severity::kERROR > nvinfer1::plugin::gLogError |
LogStream< ILogger::Severity::kWARNING > nvinfer1::plugin::gLogWarning |
LogStream< ILogger::Severity::kINFO > nvinfer1::plugin::gLogInfo |
LogStream< ILogger::Severity::kVERBOSE > nvinfer1::plugin::gLogVerbose |