Module: tfm.vision.configs.image_classification

View source on GitHub

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.

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