Skip to content

This is an unofficial implementation of BOAT: Bilateral Local Attention Vision Transformer

Notifications You must be signed in to change notification settings

Udayeon/pytorch-boat

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 

Repository files navigation

BOAT: Bilateral Local Attention Vision Transformer

This is an unofficial implementation of the paper BOAT: Bilateral Local Attention Vision Transformer.

The Swin variant is based on Swin Transformer

The CSwin variant is based on CSwin Tranformer

Please check corresponding folders for more installation, training and evaluation instructions.

Pre-trained models

BOAT-Swin-Tiny

BOAT-Swin-Small

BOAT-Swin-Base

BOAT-CSwin-Tiny

BOAT-CSwin-Small

BOAT-CSwin-Base

Acknowledgement

This is developped based on CSWin Transformer and Swin-transformer

If you use this code for your research, please consider citing:

@article{BOAT,
  author    = {Tan Yu and Gangming Zhao and Ping Li and Yizhou Yu},
  title     = {{BOAT:} Bilateral Local Attention Vision Transformer},
  journal   = {CoRR},
  volume    = {abs/2201.13027},
  year      = {2022},
  url       = {https://arxiv.org/abs/2201.13027},
}

About

This is an unofficial implementation of BOAT: Bilateral Local Attention Vision Transformer

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.4%
  • Shell 0.6%