/PyTransformer

A Collection of the Popular Vision Transformer Structure with Support for Downstream Tasks.

Primary LanguagePythonApache License 2.0Apache-2.0

PyTransformer

A Collection of the Popular Vision Transformer Structure with Support for Downstream Tasks.(Mainly for the Object Detection/Instance Segmentation Tasks)

Highlight

  • Based on the Easy-to-use cvpods
  • Support for the state-of-the-art models

Model Support

Classification

Vision Transformer Related

  • Swin-T/S/B/L
  • DeiT
  • PVT

Segmentation

Vision Transformer Related

  • Swin-T

Acknowledgement

  • cvpods
  • detectron2
  • mmdetection
  • swin-transformer