Segformer Python Packages

segmentation-models-pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

fashn-human-parser

Human parsing model for fashion and virtual try-on applications

d3m-segmentation-models-pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

segmentation-models-pytorch-deepflash2

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

deepvision-toolkit

PyTorch and TensorFlow/Keras image models with automatic weight conversions and equal API/implementations - Vision Transformer (ViT), ResNetV2, EfficientNetV2, NeRF, SegFormer, MixTransformer, (planned...) DeepLabV3+, ConvNeXtV2, YOLO, etc.

hairtone

Portrait hair re-colouring with a SegFormer ∪ BiSeNet UNION mask, region growing, and warm-shift LAB Reinhard transfer

segmentation-sust

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.