Add Ultralytics ViT Docs (#3230)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Glenn Jocher
2023-06-17 16:44:41 +02:00
committed by GitHub
parent bd0f7ecf6f
commit 1e5702a5b5
37 changed files with 451 additions and 32 deletions

View File

@ -0,0 +1,9 @@
## MaskDecoder
---
### ::: ultralytics.vit.sam.modules.decoders.MaskDecoder
<br><br>
## MLP
---
### ::: ultralytics.vit.sam.modules.decoders.MLP
<br><br>

View File

@ -0,0 +1,54 @@
---
description: Learn about Ultralytics ViT encoder, position embeddings, attention, window partition, and more in our comprehensive documentation.
keywords: Ultralytics YOLO, ViT Encoder, Position Embeddings, Attention, Window Partition, Rel Pos Encoding
---
## ImageEncoderViT
---
### ::: ultralytics.vit.sam.modules.encoders.ImageEncoderViT
<br><br>
## PromptEncoder
---
### ::: ultralytics.vit.sam.modules.encoders.PromptEncoder
<br><br>
## PositionEmbeddingRandom
---
### ::: ultralytics.vit.sam.modules.encoders.PositionEmbeddingRandom
<br><br>
## Block
---
### ::: ultralytics.vit.sam.modules.encoders.Block
<br><br>
## Attention
---
### ::: ultralytics.vit.sam.modules.encoders.Attention
<br><br>
## PatchEmbed
---
### ::: ultralytics.vit.sam.modules.encoders.PatchEmbed
<br><br>
## window_partition
---
### ::: ultralytics.vit.sam.modules.encoders.window_partition
<br><br>
## window_unpartition
---
### ::: ultralytics.vit.sam.modules.encoders.window_unpartition
<br><br>
## get_rel_pos
---
### ::: ultralytics.vit.sam.modules.encoders.get_rel_pos
<br><br>
## add_decomposed_rel_pos
---
### ::: ultralytics.vit.sam.modules.encoders.add_decomposed_rel_pos
<br><br>

View File

@ -0,0 +1,9 @@
---
description: Learn about the SamAutomaticMaskGenerator module in Ultralytics YOLO, an automatic mask generator for image segmentation.
keywords: SamAutomaticMaskGenerator, Ultralytics YOLO, automatic mask generator, image segmentation
---
## SamAutomaticMaskGenerator
---
### ::: ultralytics.vit.sam.modules.mask_generator.SamAutomaticMaskGenerator
<br><br>

View File

@ -0,0 +1,9 @@
---
description: Learn about PromptPredictor - a module in Ultralytics VIT SAM that predicts image captions based on prompts. Get started today!.
keywords: PromptPredictor, Ultralytics, YOLO, VIT SAM, image captioning, deep learning, computer vision
---
## PromptPredictor
---
### ::: ultralytics.vit.sam.modules.prompt_predictor.PromptPredictor
<br><br>

View File

@ -0,0 +1,9 @@
---
description: Explore the Sam module in Ultralytics VIT, a PyTorch-based vision library, and learn how to improve your image classification and segmentation tasks.
keywords: Ultralytics VIT, Sam module, PyTorch vision library, image classification, segmentation tasks
---
## Sam
---
### ::: ultralytics.vit.sam.modules.sam.Sam
<br><br>

View File

@ -0,0 +1,19 @@
---
description: Explore the Attention and TwoWayTransformer modules in Ultralytics YOLO documentation. Learn how to integrate them in your project efficiently.
keywords: Ultralytics YOLO, Attention module, TwoWayTransformer module, Object Detection, Deep Learning
---
## TwoWayTransformer
---
### ::: ultralytics.vit.sam.modules.transformer.TwoWayTransformer
<br><br>
## TwoWayAttentionBlock
---
### ::: ultralytics.vit.sam.modules.transformer.TwoWayAttentionBlock
<br><br>
## Attention
---
### ::: ultralytics.vit.sam.modules.transformer.Attention
<br><br>