class

CSPNet

extendsPretrainedModelBackboneMixin

CSPNet(config: CSPNetConfig)

CSPNet feature-extracting backbone (Wang et al., CVPRW 2020).

Builds a four-stage CSP-wrapped (or plain Darknet) trunk according to CSPNetConfig. Returns the final feature map without pooling.

Constructors

dunder

→None

__init__(config: CSPNetConfig)

prop

→list[FeatureInfo]

feature_info: list[FeatureInfo]

→BaseModelOutput

forward(x: Tensor)

→Tensor

forward_features(x: Tensor)