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