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