ConvNeXt
18 memberslucid.models.vision.convnextConvNeXt family — Liu et al., 2022.
Liu, Zhuang, et al. "A ConvNet for the 2020s." Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2022, pp. 11976-11986.
ConvNeXt is a pure convolutional architecture obtained by systematically modernizing a ResNet using design choices borrowed from the Swin Transformer. Starting from a ResNet-50 the authors apply a one-at-a-time roadmap: change the stage compute ratio to , swap the stem for a non-overlapping patchify convolution, replace the bottleneck with a depthwise convolution followed by an inverted bottleneck, move the activation and normalization to a single GELU + LayerNorm after the depthwise conv, and replace BatchNorm with LayerNorm. The resulting block is
where is a layer-scale parameter initialized to (Touvron et al., 2021).
The pyramid follows the four-stage Swin layout — channel widths with downsampling between stages — and the canonical variants Tiny / Small / Base / Large / XLarge scale the stem width and the depth of stage 3 from to . The paper's headline result is that the modernised ConvNet matches or exceeds Swin's ImageNet accuracy at the same FLOPs while remaining a simple all-convolutional design.
Classes
Functions
convnext_tiny→ ConvNeXtConvNeXt-Tiny backbone (Liu et al., 2022).
convnext_tiny_cls→ ConvNeXtForImageClassificationConvNeXt-Tiny image classifier (Liu et al., 2022).
convnext_small→ ConvNeXtConvNeXt-Small backbone (Liu et al., 2022).
convnext_small_cls→ ConvNeXtForImageClassificationConvNeXt-Small image classifier (Liu et al., 2022).
convnext_base→ ConvNeXtConvNeXt-Base backbone (Liu et al., 2022).
convnext_base_cls→ ConvNeXtForImageClassificationConvNeXt-Base image classifier (Liu et al., 2022).
convnext_large→ ConvNeXtConvNeXt-Large backbone (Liu et al., 2022).
convnext_large_cls→ ConvNeXtForImageClassificationConvNeXt-Large image classifier (Liu et al., 2022).
convnext_xlarge→ ConvNeXtConvNeXt-XLarge backbone (Liu et al., 2022).
convnext_xlarge_cls→ ConvNeXtForImageClassificationConvNeXt-XLarge image classifier (Liu et al., 2022).
Weights
ConvNeXtTinyWeightsPretrained weights for lucid.models.convnext_tiny_cls.
ConvNeXtSmallWeightsPretrained weights for lucid.models.convnext_small_cls.
ConvNeXtBaseWeightsPretrained weights for lucid.models.convnext_base_cls.
ConvNeXtLargeWeightsPretrained weights for lucid.models.convnext_large_cls.
ConvNeXtXLargeWeightsPretrained weights for lucid.models.convnext_xlarge_cls.