Swin Transformer TensorFlow Implementation

• Swin-Transformer-Tensorflow

  2 61 5.0 Python

  Unofficial implementation of "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" (https://arxiv.org/abs/2103.14030)

  

• Swin-Transformer

  23 12,975 2.8 Python

  This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

  

Hey guys! A few others and I recently implemented a TensorFlow version of Microsoft's Swin Transformer (https://arxiv.org/abs/2103.14030). It's an almost direct translation of the official PyTorch code so that people can easily switch reading between the two. The TensorFlow version repo also includes code that converts PyTorch .pth weights into TensorFlow checkpoints. Using this file, you guys can either use the pretrained weights provided by Microsoft or weights of a custom model trained using PyTorch. I hope you guys find it useful! Here's the GitHub link:

• WorkOS

  workos.com sponsored

  The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.

  

Suggest a related project