任务
BASNet分割
DeepLabV3Plus分割
SegFormer分割模型
- SegFormer class
- from_preset method
- SegFormerB0 class
- SegFormerB1 class
- SegFormerB2 class
- SegFormerB3 class
- SegFormerB4 class
- SegFormerB5 class
分割任何物体
- SegmentAnythingModel class
- from_preset method
- SAMMaskDecoder class
- SAMPromptEncoder class
- TwoWayTransformer class
- MultiHeadAttentionWithDownsampling class
- TwoWayMultiHeadAttention class
- RandomFrequencyPositionalEmbeddings class
CLIP 特征提取器
- CLIP class
- from_preset method
- CLIPAttention class
- CLIPEncoder class
- CLIPImageEncoder class
- CLIPProcessor class
- CLIPTextEncoder class
- QuickGELU class
- ResidualAttention class
图像分类器模型
RetinaNet模型
StableDiffusion 图像生成模型
- StableDiffusion class
- StableDiffusionV2 class
- Decoder class
- DiffusionModel class
- ImageEncoder class
- NoiseScheduler class
- SimpleTokenizer class
- TextEncoder class
- TextEncoderV2 class