transform_anchors

This plugin runs the transform_anchors algorithm. It requires two inputs, classification and regression tensors; and creates tensor of BBoxes with coordinates, class_id and score. At this time this plugin is specific to the EfficientDet model.

Features

  • Performs ArgMax + Sigmoid + Filter by threshold on classification tensor

  • Uses NEON optimized ArgMax implementation

Usage

1 simaaisrc location=/data/clipped_boxes.bin node-name="boxes" ! simaaitransformanchors name=transformanchors config='/data/simaai/applications/EfficientDet/etc/transform_anchors_config.json' ! ... simaaisrc location=/data/classification.bin !  transformanchors.

Configuration

This plugin is specific to the EfficientDet model and therefore not configurable at this time.