任務
BASNet 分割
DeepLabV3Plus 分割
SegFormer 分割
- SegFormer 類別
- from_preset 方法
- SegFormerB0 類別
- SegFormerB1 類別
- SegFormerB2 類別
- SegFormerB3 類別
- SegFormerB4 類別
- SegFormerB5 類別
Segment Anything
- SegmentAnythingModel 類別
- from_preset 方法
- SAMMaskDecoder 類別
- SAMPromptEncoder 類別
- TwoWayTransformer 類別
- MultiHeadAttentionWithDownsampling 類別
- TwoWayMultiHeadAttention 類別
- RandomFrequencyPositionalEmbeddings 類別
CLIP 特徵提取器
- CLIP 類別
- from_preset 方法
- CLIPAttention 類別
- CLIPEncoder 類別
- CLIPImageEncoder 類別
- CLIPProcessor 類別
- CLIPTextEncoder 類別
- QuickGELU 類別
- ResidualAttention 類別
ImageClassifier 模型
RetinaNet 模型
StableDiffusion 圖像生成模型
- StableDiffusion 類別
- StableDiffusionV2 類別
- Decoder 類別
- DiffusionModel 類別
- ImageEncoder 類別
- NoiseScheduler 類別
- SimpleTokenizer 類別
- TextEncoder 類別
- TextEncoderV2 類別