 |
TensorRT
7.2.1.6
NVIDIA TensorRT
|
|
|
TensorRT
7.2.1.6
NVIDIA TensorRT
|
|
Table Of Contents
The generateDetection plugin performs bounding boxe refinement of MaskRCNN's detection head and generates the final detection output of MaskRCNN.
This plugin supports the NCHW format. It takes three input tensors: delta_bbox, score and roi
delta_bbox is the refinement information of roi boxes generated from the MultilevelProposeROI plugin. delta_bbox tensor's shape is [N, rois, num_classes*4, 1, 1] where N is batch size, rois is the total number of ROI boxes candidates per image, and num_classes*4 means 4 refinement elements ([dy, dx, dh, dw]) for each roi box as different classes.
score is the predicted class scores of ROI boxes generated from MaskRCNN detection head of shape [N, rois, num_classes, 1, 1]. There is an argmax operation in generateDetection to determine the final class of detection candidates.
roi is the coordinates of ROI boxes candidates from the MultilevelProposeROI plugin of shape [N, rois, 4].
This plugin generates output of shape [N, keep_topk, 6] where keep_topk is the maximum number of detections left after NMS and '6' means 6 elements of an detection [y1, x1, y2, x2, class_label, score]
This plugin has the plugin creator class generateDetectionPluginCreator and the plugin class generateDetection.
The following parameters were used to create generateDetection instance:
| Type | Parameter | Description |
|---|---|---|
int | num_classes | Number of detection classes(including background). num_classes=91 for COCO dataset |
int | keep_topk | Number of detections will be kept after NMS. |
float | score_threshold | Confidence threshold value. This plugin will drop a detection if its class confidence(score) is under "score_threshold". |
float | iou_threshold | IOU threshold value used in NMS. |
For terms and conditions for use, reproduction, and distribution, see the TensorRT Software License Agreement documentation.
June 2020 This is the first release of this README.md file.
There are no known issues in this plugin.
1.8.17