vis4d.op.fpp.yolox_pafpn

YOLOX PAFPN.

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

Classes

YOLOXPAFPN(in_channels, out_channels[, ...])

Path Aggregation Network used in YOLOX.
class YOLOXPAFPN(in_channels, out_channels, num_csp_blocks=3, start_index=2)[source]

Path Aggregation Network used in YOLOX.

Parameters:

  • in_channels (list[int]) – Number of input channels per scale.

  • out_channels (int) – Number of output channels (used at each scale).

  • num_csp_blocks (int, optional) – Number of bottlenecks in CSPLayer. Defaults to 3.

  • start_index (int, optional) – Index of the first input feature map. Defaults to 2.

Init.

__call__(features)[source]

Type definition for call implementation.

Return type:

list[Tensor]

forward(features)[source]

Forward pass.

Parameters:

features (tuple[Tensor]) – Input features.

Returns:

YOLOXPAFPN features.

Return type:

list[Tensor]