vis4d.op.box.encoder.yolox

YOLOX decoder for 2D boxes.

Modified from mmdetection (https://github.com/open-mmlab/mmdetection).

Classes

YOLOXBBoxDecoder()

YOLOX BBox decoder.
class YOLOXBBoxDecoder[source]

YOLOX BBox decoder.

__call__(points, offsets)[source]

Apply box offsets to points, used by YOLOX.

Parameters:

  • points (Tensor) – Points. Shape (B, N, 4) or (N, 4).

  • offsets (Tensor) – Offsets. Has shape (B, N, 4) or (N, 4).

Returns:

Decoded boxes.

Return type:

Tensor