Module: tfm.vision.configs.image_classification

Image classification configuration definition.

Classes

class DataConfig: Input config for training.

class Evaluation

class ImageClassificationModel: The model config.

class ImageClassificationTask: The task config.

class Losses

Functions

image_classification(...): Image classification general.

image_classification_imagenet(...): Image classification on imagenet with resnet.

image_classification_imagenet_deit_pretrain(...): Image classification on imagenet with vision transformer.

image_classification_imagenet_mobilenet(...): Image classification on imagenet with mobilenet.

image_classification_imagenet_resnetrs(...): Image classification on imagenet with resnet-rs.

image_classification_imagenet_revnet(...): Returns a revnet config for image classification on imagenet.

image_classification_imagenet_vit_finetune(...): Image classification on imagenet with vision transformer.

image_classification_imagenet_vit_pretrain(...): Image classification on imagenet with vision transformer.

IMAGENET_INPUT_PATH_BASE 'imagenet-2012-tfrecord' IMAGENET_TRAIN_EXAMPLES 1281167 IMAGENET_VAL_EXAMPLES 50000