Tasks
BASNet Segmentation
DeepLabV3Plus Segmentation
SegFormer Segmentation
- SegFormer class
- from_preset method
- SegFormerB0 class
- SegFormerB1 class
- SegFormerB2 class
- SegFormerB3 class
- SegFormerB4 class
- SegFormerB5 class
Segment Anything
- SegmentAnythingModel class
- from_preset method
- SAMMaskDecoder class
- SAMPromptEncoder class
- TwoWayTransformer class
- MultiHeadAttentionWithDownsampling class
- TwoWayMultiHeadAttention class
- RandomFrequencyPositionalEmbeddings class
CLIP Feature extractor
- CLIP class
- from_preset method
- CLIPAttention class
- CLIPEncoder class
- CLIPImageEncoder class
- CLIPProcessor class
- CLIPTextEncoder class
- QuickGELU class
- ResidualAttention class
The ImageClassifier model
The RetinaNet model
StableDiffusion image-generation model
- StableDiffusion class
- StableDiffusionV2 class
- Decoder class
- DiffusionModel class
- ImageEncoder class
- NoiseScheduler class
- SimpleTokenizer class
- TextEncoder class
- TextEncoderV2 class