summaryrefslogtreecommitdiff
path: root/candle-transformers/src/models/segment_anything/mod.rs
diff options
context:
space:
mode:
Diffstat (limited to 'candle-transformers/src/models/segment_anything/mod.rs')
-rw-r--r--candle-transformers/src/models/segment_anything/mod.rs8
1 files changed, 8 insertions, 0 deletions
diff --git a/candle-transformers/src/models/segment_anything/mod.rs b/candle-transformers/src/models/segment_anything/mod.rs
index c54493d2..3e85fe35 100644
--- a/candle-transformers/src/models/segment_anything/mod.rs
+++ b/candle-transformers/src/models/segment_anything/mod.rs
@@ -1,3 +1,11 @@
+//! Segment Anything Model (SAM)
+//!
+//! SAM is an architecture for image segmentation, capable of segmenting any object
+//! in an image based on prompts like points or boxes.
+//!
+//! - [GH Link](https://github.com/facebookresearch/segment-anything)
+//! - [Paper](https://arxiv.org/abs/2304.02643)
+//!
pub use crate::models::with_tracing::Linear;
use candle::{Result, Tensor};
use candle_nn::{Module, VarBuilder};