SwAV-TF

TensorFlow implementation of "Unsupervised Learning of Visual Features by Contrasting Cluster Assignments".

APACHE-2.0 License

Stars

Ecosystems: Keras, Tensorflow

Statistics for this project are still being loaded, please check back later.

Badges

Extracted from project README

Learning Tensorflow Step by Step:: Concepts, Examples & Applications

Unsupervised Data Augmentation (UDA)

Implements the ideas presented in https://arxiv.org/pdf/2004.11362v1.pdf by Khosla et al.

Custom Keras Callbacks for Feature Visualization, Class Activation Map, Grad-CAM

(Minimally) implements SimCLR (https://arxiv.org/abs/2002.05709) in TensorFlow 2.

Implementation of CaiT models in TensorFlow and ImageNet-1k checkpoints. Includes code for infere...

Includes additional materials for the following keras.io blog post.

Minimal implementation of PAWS (https://arxiv.org/abs/2104.13963) in TensorFlow.

Implementation of Swin Transformers in TensorFlow along with converted pre-trained models, code f...

This repository hosts code for converting the original Vision Transformer models (JAX) to TensorF...

Contains code for the paper "Vision Transformers are Robust Learners" (AAAI 2022).