GitHub - martinsbruveris/tensorflow-image-models: TensorFlow port of PyTorch Image Models (timm) - image models with pretrained weights.

TensorFlow Image Models (tfimm) is a collection of image models with pretrained weights, obtained by porting architectures from timm to TensorFlow. The hope is that the number of available architectures will grow over time. For now, it contains vision transformers (ViT, DeiT, CaiT, PVT and Swin Transformers), MLP-Mixer models (MLP-Mixer, ResMLP, gMLP, PoolFormer and ConvMixer), various ResNet flav

[misshiki]

timm（PyTorch Image Models）をTensorFlowに移植したPythonライブラリのtfimm（「T」ensor「F」low 「Im」age 「M」odels）。コンピュータビジョン系の事前訓練済みモデルが使える。