TensorFlow Image Models (tfimm)

TensorFlow Image Models (tfimm) is a collection of image models with pretrained weights, obtained by porting architectures from timm to TensorFlow. It contains vision transformers (ViT, DeiT, CaiT, PVT and Swin Transformers), MLP-Mixer models (MLP-Mixer, ResMLP, gMLP and ConvMixer) and various ResNet flavours (ResNet, ResNeXt, ECA-ResNet, SE-ResNet).

Contents

Transformer Models

Configuration and parameters specific to transformers

Trainer Class

Trainer class

Library

Architectures

Training

Applications