stouputils.data_science.models.all module#

class ModelClassMap[source]#

Bases: dict[type[ModelInterface], tuple[str, …]]

Dictionary mapping class to their names and aliases.

  • SqueezeNet: (‘squeezenet’, ‘squeezenets’, ‘all’, ‘often’)

  • DenseNet121: (‘densenet121’, ‘densenets’, ‘all’, ‘often’, ‘good’)

  • DenseNet169: (‘densenet169’, ‘densenets’, ‘all’, ‘often’, ‘good’)

  • DenseNet201: (‘densenet201’, ‘densenets’, ‘all’, ‘often’, ‘good’)

  • EfficientNetB0: (‘efficientnetb0’, ‘efficientnets’, ‘all’)

  • EfficientNetV2B0: (‘efficientnetv2b0’, ‘efficientnets’, ‘all’)

  • EfficientNetV2S: (‘efficientnetv2s’, ‘efficientnets’, ‘all’, ‘often’)

  • EfficientNetV2M: (‘efficientnetv2m’, ‘efficientnets’, ‘all’, ‘often’)

  • EfficientNetV2L: (‘efficientnetv2l’, ‘efficientnets’, ‘all’, ‘often’)

  • ConvNeXtTiny: (‘convnexttiny’, ‘convnexts’, ‘all’, ‘often’, ‘good’)

  • ConvNeXtSmall: (‘convnextsmall’, ‘convnexts’, ‘all’, ‘often’)

  • ConvNeXtBase: (‘convnextbase’, ‘convnexts’, ‘all’, ‘often’, ‘good’)

  • ConvNeXtLarge: (‘convnextlarge’, ‘convnexts’, ‘all’, ‘often’)

  • ConvNeXtXLarge: (‘convnextxlarge’, ‘convnexts’, ‘all’, ‘often’, ‘good’)

  • VGG16: (‘vgg16’, ‘vggs’, ‘all’)

  • VGG19: (‘vgg19’, ‘vggs’, ‘all’)

  • MobileNet: (‘mobilenet’, ‘mobilenets’, ‘all’)

  • MobileNetV2: (‘mobilenetv2’, ‘mobilenets’, ‘all’, ‘often’)

  • MobileNetV3Small: (‘mobilenetv3small’, ‘mobilenets’, ‘all’, ‘often’)

  • MobileNetV3Large: (‘mobilenetv3large’, ‘mobilenets’, ‘all’, ‘often’, ‘good’)

  • ResNet50V2: (‘resnet50v2’, ‘resnetsv2’, ‘resnets’, ‘all’, ‘often’)

  • ResNet101V2: (‘resnet101v2’, ‘resnetsv2’, ‘resnets’, ‘all’, ‘often’)

  • ResNet152V2: (‘resnet152v2’, ‘resnetsv2’, ‘resnets’, ‘all’, ‘often’)

  • Xception: (‘xception’, ‘xceptions’, ‘all’, ‘often’)

  • Sandbox: (‘sandbox’,)

ALL_MODELS: list[str] = ['all', 'convnextbase', 'convnextlarge', 'convnexts', 'convnextsmall', 'convnexttiny', 'convnextxlarge', 'densenet121', 'densenet169', 'densenet201', 'densenets', 'efficientnetb0', 'efficientnets', 'efficientnetv2b0', 'efficientnetv2l', 'efficientnetv2m', 'efficientnetv2s', 'good', 'mobilenet', 'mobilenets', 'mobilenetv2', 'mobilenetv3large', 'mobilenetv3small', 'often', 'resnet101v2', 'resnet152v2', 'resnet50v2', 'resnets', 'resnetsv2', 'sandbox', 'squeezenet', 'squeezenets', 'vgg16', 'vgg19', 'vggs', 'xception', 'xceptions']#

All models names and aliases found in the CLASS_MAP dictionary.

On this page

This Page