Module convnext

Module convnext 

Source
Expand description

ConvNeXt implementation.

This candle implementation uses a pre-trained ConvNeXt network for inference. The classification head has been trained on the ImageNet dataset and returns the probabilities for the top-5 classes.

Original code:

  • ๐Ÿ’ป ConvNeXt
  • ๐Ÿ’ป ConvNeXt-V2
  • ๐Ÿ’ป timm
  • ๐Ÿ“ Paper A ConvNet for the 2020s
  • ๐Ÿ“ Paper ConvNeXt V2: Co-designing and Scaling ConvNets with Masked Autoencoders

Structsยง

Config

Functionsยง

convnext
convnext_no_final_layer