Table Of Contents
The SpecialSlice
plugin slice the detections of MaskRCNN from [y1, x1, y2, x2, class_label, score]
to [y1, x1, y2, x2]
. It is used in sampleMaskRCNN.
This plugin supports the NCHW format. It takes one input tensor: detections
detections
is the output of DetectionLayer
in MaskRCNN model. Its shape is [N, num_det, 6]
where N
is the batch size, num_det
is the number of detections generated from DetectionLayer
and 6
means 6 elements of a detection [y1, x1, y2, x2, class_label, score]
.
This plugin generates one output tensor of shape [N, num_det, 4]
.
This plugin has the plugin creator class FlattenConcatPluginCreator
and the plugin class FlattenConcat
.
This plugin has no parameter.
The following resources provide a deeper understanding of the flattenConcat
plugin:
For terms and conditions for use, reproduction, and distribution, see the TensorRT Software License Agreement documentation.
June 2019 This is the first release of this README.md
file.
There are no known issues in this plugin.