Function sine_encode_boxes

tests/dump_geom_encoder_reference.py:107–111

Encode box (cx, cy, w, h) with sinusoidal PE. Returns [N, 258].

(cx, cy, w, h, num_pos_feats=128, temperature=10000)

def sine_encode_boxes(cx, cy, w, h, num_pos_feats=128, temperature=10000):
    """Encode box (cx, cy, w, h) with sinusoidal PE. Returns [N, 258]."""
    pos_x, pos_y = sine_encode_xy(cx, cy, num_pos_feats, temperature)
    pos = torch.cat((pos_y, pos_x, h[:, None], w[:, None]), dim=1)  # [N, 258]
    return pos


# ── Box coordinate conversion ──────────────────────────────────────────────