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_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.

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