SSD_TO_BBOX

SSDToBoundingBoxStep is a pipeline step that configures extraction of bounding boxes from a (Single-Shot Detector) SSD model output.

Configs

Descriptions

classLabels

A list of class labels

keepOtherValues

true by default. If true, other data key and values from the previous step are kept and passed on to the next step as well.

threshold

Threadshold to the output of the SSD models for fetching bounding boxes for. Default threshold is 0.5.

scale

An optional way to increase the size of the bounding boxes by some fraction. If specified, a value of 1.0 is equivalent to no scaling. A scale of 2.0 means the center is unchanged, but the width and height are now two times larger than it would be.

aspectRatio

An optional way to control the output shape (aspect ratio) of the bounding boxes. Defined in terms of width or height. If specified, an aspect ratio of 1.0 gives a square output; an aspect ratio of 2.0 gives twice as wide as it is high. Note that for making the output the correct aspect ratio, one of the height or width will be increased; the other dimension will not change. The pre-aspect-ratio-corrected box will be contained fully within the output box.

outputName

Output key name where the bounding box will be contained in. The default name is bounding_box.

Last updated