sk_resnet_50_cls(pretrained: bool = False, overrides: object = {})SK-ResNet-50 image classifier (backbone + GAP + linear head).
Builds an SKNetForImageClassification with the
SK-ResNet-50 backbone (bottleneck blocks stacked
[3, 4, 6, 3], one SK unit per bottleneck) followed by
global average pooling and a linear projection. Approximately
25.8M parameters.
Model Size
Parameters
pretrainedbool= FalseReserved for future pretrained-weight loading. Currently
ignored.
**overridesobject= {}Keyword overrides forwarded into
SKNetConfig.Returns
SKNetForImageClassificationClassifier with the SK-ResNet-50 configuration applied
(or with overrides merged on top of it).
Notes
See Li et al., "Selective Kernel Networks", CVPR 2019 (arXiv:1903.06586), Table 1.
Examples
>>> import lucid
>>> from lucid.models.vision.sknet import sk_resnet_50_cls
>>> model = sk_resnet_50_cls(num_classes=10)
>>> x = lucid.randn(2, 3, 224, 224)
>>> out = model(x)
>>> out.logits.shape
(2, 10)