MaxViT-Base backbone (Tu et al., 2022).
Builds the canonical MaxViT-Base configuration:
depths=(2, 6, 14, 2), dims=(96, 192, 384, 768).
Approximately 96.6M parameters.
Model Size
Parameters
pretrainedbool= FalseIf
True, loads ImageNet-1k or ImageNet-22k pretrained
weights when available. Defaults to False.**overridesobject= {}Keyword overrides on top of the canonical MaxViT-Base config.
Returns
MaxViTA MaxViT backbone returning a flat
feature.
Notes
MaxViT-Base reaches 84.9% top-1 on ImageNet-1k at 224x224 (Tu et al., 2022, Table 6).
Examples
>>> import lucid
>>> from lucid.models.vision.maxvit import maxvit_base
>>> model = maxvit_base()
>>> x = lucid.randn(1, 3, 224, 224)
>>> model.forward_features(x).shape
(1, 768)