TLTMaskRCNNConfig Namespace Reference

Variables
static const nvinfer1::Dims3	IMAGE_SHAPE {3, 832, 1344}
static const int	POOL_SIZE = 7
static const int	MASK_POOL_SIZE = 14
static const float	MASK_THRESHOLD = 0.5
static const float	DETECTION_REG_WEIGHTS [] = {10, 10, 5, 5}
static const int	DETECTION_MAX_INSTANCES = 100
static const float	DETECTION_MIN_CONFIDENCE = 0
static const float	DETECTION_NMS_THRESHOLD = 0.5
static const int	FPN_CLASSIF_FC_LAYERS_SIZE = 1024
static const int	TOP_DOWN_PYRAMID_SIZE = 256
static const int	NUM_CLASSES = 1 + 90
static const int	MIN_LEVEL = 2
static const int	MAX_LEVEL = 6
static const float	RPN_ANCHOR_SCALE = 8
static const std::vector< std::pair< float, float > >	ANCHOR_RATIOS = {std::make_pair(1.0f, 1.0f), std::make_pair(1.4f, 0.7f), std::make_pair(0.7f, 1.4f)}
static const int	RPN_ANCHOR_STRIDE = 1
static const int	MAX_PRE_NMS_RESULTS = 1000
static const float	RPN_NMS_THRESHOLD = 0.7
static const int	POST_NMS_ROIS_INFERENCE = 1000
static const std::vector< std::string >	CLASS_NAMES
static const std::string	MODEL_NAME = "mrcnn_nchw.uff"
static const std::string	MODEL_INPUT = "Input"
static const Dims3	MODEL_INPUT_SHAPE = IMAGE_SHAPE
static const std::vector< std::string >	MODEL_OUTPUTS = {"generate_detections", "mask_head/mask_fcn_logits/BiasAdd"}
static const Dims2	MODEL_DETECTION_SHAPE {DETECTION_MAX_INSTANCES, 6}
static const Dims4	MODEL_MASK_SHAPE {DETECTION_MAX_INSTANCES, NUM_CLASSES, 28, 28}

Variable Documentation

IMAGE_SHAPE

const nvinfer1::Dims3 TLTMaskRCNNConfig::IMAGE_SHAPE {3, 832, 1344}

static

POOL_SIZE

const int TLTMaskRCNNConfig::POOL_SIZE = 7

static

MASK_POOL_SIZE

const int TLTMaskRCNNConfig::MASK_POOL_SIZE = 14

static

MASK_THRESHOLD

const float TLTMaskRCNNConfig::MASK_THRESHOLD = 0.5

static

DETECTION_REG_WEIGHTS

const float TLTMaskRCNNConfig::DETECTION_REG_WEIGHTS[] = {10, 10, 5, 5}

static

DETECTION_MAX_INSTANCES

const int TLTMaskRCNNConfig::DETECTION_MAX_INSTANCES = 100

static

DETECTION_MIN_CONFIDENCE

const float TLTMaskRCNNConfig::DETECTION_MIN_CONFIDENCE = 0

static

DETECTION_NMS_THRESHOLD

const float TLTMaskRCNNConfig::DETECTION_NMS_THRESHOLD = 0.5

static

FPN_CLASSIF_FC_LAYERS_SIZE

const int TLTMaskRCNNConfig::FPN_CLASSIF_FC_LAYERS_SIZE = 1024

static

TOP_DOWN_PYRAMID_SIZE

const int TLTMaskRCNNConfig::TOP_DOWN_PYRAMID_SIZE = 256

static

NUM_CLASSES

const int TLTMaskRCNNConfig::NUM_CLASSES = 1 + 90

static

MIN_LEVEL

const int TLTMaskRCNNConfig::MIN_LEVEL = 2

static

MAX_LEVEL

const int TLTMaskRCNNConfig::MAX_LEVEL = 6

static

RPN_ANCHOR_SCALE

const float TLTMaskRCNNConfig::RPN_ANCHOR_SCALE = 8

static

ANCHOR_RATIOS

const std::vector<std::pair<float, float> > TLTMaskRCNNConfig::ANCHOR_RATIOS = {std::make_pair(1.0f, 1.0f), std::make_pair(1.4f, 0.7f), std::make_pair(0.7f, 1.4f)}

static

RPN_ANCHOR_STRIDE

const int TLTMaskRCNNConfig::RPN_ANCHOR_STRIDE = 1

static

MAX_PRE_NMS_RESULTS

const int TLTMaskRCNNConfig::MAX_PRE_NMS_RESULTS = 1000

static

RPN_NMS_THRESHOLD

const float TLTMaskRCNNConfig::RPN_NMS_THRESHOLD = 0.7

static

POST_NMS_ROIS_INFERENCE

const int TLTMaskRCNNConfig::POST_NMS_ROIS_INFERENCE = 1000

static

CLASS_NAMES

const std::vector<std::string> TLTMaskRCNNConfig::CLASS_NAMES

static

MODEL_NAME

const std::string TLTMaskRCNNConfig::MODEL_NAME = "mrcnn_nchw.uff"

static

MODEL_INPUT

const std::string TLTMaskRCNNConfig::MODEL_INPUT = "Input"

static

MODEL_INPUT_SHAPE

const Dims3 TLTMaskRCNNConfig::MODEL_INPUT_SHAPE = IMAGE_SHAPE

static

MODEL_OUTPUTS

const std::vector<std::string> TLTMaskRCNNConfig::MODEL_OUTPUTS = {"generate_detections", "mask_head/mask_fcn_logits/BiasAdd"}

static

MODEL_DETECTION_SHAPE

const Dims2 TLTMaskRCNNConfig::MODEL_DETECTION_SHAPE {DETECTION_MAX_INSTANCES, 6}

static

MODEL_MASK_SHAPE

const Dims4 TLTMaskRCNNConfig::MODEL_MASK_SHAPE {DETECTION_MAX_INSTANCES, NUM_CLASSES, 28, 28}

static