SushiAI::Detect

constexpr float K_GRID_CELL_OFFSET = 0.5f

Holds the distance from a cell's index to its centre, in grid units.

constexpr std::size_t K_BOX_VALUES = 4

Holds the number of values in one box.

constexpr float K_IOU_EPS = 1e-7f

Holds the term that keeps the IoU divisions away from zero.

std::int64_t count_anchors(SushiRuntime::span< const FeatureLevel > levels)

Returns the number of anchors over levels, one per cell.

void make_anchors(SushiRuntime::span< const FeatureLevel > levels, SushiRuntime::span< float > points, SushiRuntime::span< float > strides)

Writes each anchor's cell centre in grid units and its stride.

void scale_points(SushiRuntime::span< const float > points, SushiRuntime::span< const float > strides, SushiRuntime::span< float > scaled)

Multiplies each anchor point by its stride, which gives the point in pixels.

void decode_xyxy(SushiRuntime::span< const float > distances, SushiRuntime::span< const float > anchor_points, SushiRuntime::span< float > boxes)

Decodes [B, A, 4] distances about [A, 2] anchor points into x1 y1 x2 y2.

void decode_xywh(SushiRuntime::span< const float > distances, SushiRuntime::span< const float > anchor_points, SushiRuntime::span< float > boxes)

Decodes [B, A, 4] distances about [A, 2] anchor points into centre and size.

void encode_distances(SushiRuntime::span< const float > anchor_points, SushiRuntime::span< const float > boxes, SushiRuntime::span< float > distances)

Encodes [B, A, 4] x1 y1 x2 y2 boxes as distances about [A, 2] anchor points.

void scale_boxes(SushiRuntime::span< const float > boxes, SushiRuntime::span< const float > strides, SushiRuntime::span< float > scaled)

Multiplies [B, A, 4] boxes by the [A] stride of their anchor.

void xyxy_to_xywh(SushiRuntime::span< const float > boxes, SushiRuntime::span< float > converted)

Converts [N, 4] boxes from x1 y1 x2 y2 to centre x, centre y, width, height.

void xywh_to_xyxy(SushiRuntime::span< const float > boxes, SushiRuntime::span< float > converted)

Converts [N, 4] boxes from centre x, centre y, width, height to x1 y1 x2 y2.

float intersection_over_union(SushiRuntime::span< const float > box1, SushiRuntime::span< const float > box2) noexcept

Returns the IoU of two boxes.

float complete_iou(SushiRuntime::span< const float > box1, SushiRuntime::span< const float > box2) noexcept

Returns the complete IoU of two boxes, which is at most their IoU.

void elementwise_iou(SushiRuntime::span< const float > boxes1, SushiRuntime::span< const float > boxes2, SushiRuntime::span< float > ious)

Writes the IoU of row i of boxes1 with row i of boxes2.

void elementwise_complete_iou(SushiRuntime::span< const float > boxes1, SushiRuntime::span< const float > boxes2, SushiRuntime::span< float > ious)

Writes the complete IoU of row i of boxes1 with row i of boxes2.

void pairwise_iou(SushiRuntime::span< const float > boxes1, SushiRuntime::span< const float > boxes2, SushiRuntime::span< float > ious)

Writes the IoU of every row of boxes1 with every row of boxes2.

void join_level(SushiRuntime::span< const float > level, const LevelPlacement &placement, SushiRuntime::span< float > joined)

Copies one level's [B, n, C] rows into their place in a [B, A, C] array.

void split_level(SushiRuntime::span< const float > joined, const LevelPlacement &placement, SushiRuntime::span< float > level)

Copies one level's rows out of a [B, A, C] array into [B, n, C].