vis4d.op.fpp.base

Feature pyramid processing base class.

Classes

FeaturePyramidProcessing(*args, **kwargs)

Base Neck class.
class FeaturePyramidProcessing(*args, **kwargs)[source]

Base Neck class.

Initialize internal Module state, shared by both nn.Module and ScriptModule.

__call__(features)[source]

Type definition for call implementation.

Return type:

list[Tensor]

abstract forward(features)[source]

Feature pyramid processing.

This module do a further processing for the hierarchical feature representation extracted by the base models.

Parameters:

  • features (list[Tensor]) – Feature pyramid as outputs of the

  • model. (base)

Returns:

Feature pyramid after the processing.

Return type:

list[Tensor]