Struct SegmentAnything 

pub struct SegmentAnything { /* private fields */ }

The segment anything model.

Implementations

impl SegmentAnything

pub fn builder() -> SegmentAnythingBuilder

Creates a new SegmentAnythingBuilder.

Examples found in repository

examples/segment-everything.rs (line 4)

fn main() {
    let model = SegmentAnything::builder().build().unwrap();
    let image = image::open("examples/landscape.jpg").unwrap();
    let images = model.segment_everything(image).unwrap();
    for (i, img) in images.iter().enumerate() {
        img.save(format!("{}.png", i)).unwrap();
    }
}

examples/segment-anything.rs (line 4)

fn main() {
    let model = SegmentAnything::builder().build().unwrap();
    let image = image::open("examples/landscape.jpg").unwrap();
    let x = image.width() / 2;
    let y = image.height() / 4;
    let images = model
        .segment_from_points(SegmentAnythingInferenceSettings::new(image).add_goal_point(x, y))
        .unwrap();

    images.save("out.png").unwrap();
}

pub fn segment_from_points( &self, settings: SegmentAnythingInferenceSettings, ) -> Result<DynamicImage, SegmentAnythingInferenceError>

Segment an image from a list of points. Returns a DynamicImage mask.

Example

use segment_anything_rs::*;

let model = SegmentAnything::builder().build().unwrap();
let image = image::open("examples/landscape.jpg").unwrap();
let x = image.width() / 2;
let y = image.height() / 4;
let images = model
    .segment_from_points(SegmentAnythingInferenceSettings::new(image).add_goal_point(x, y))
    .unwrap();

images.save("out.png").unwrap();

Examples found in repository

examples/segment-anything.rs (line 9)

fn main() {
    let model = SegmentAnything::builder().build().unwrap();
    let image = image::open("examples/landscape.jpg").unwrap();
    let x = image.width() / 2;
    let y = image.height() / 4;
    let images = model
        .segment_from_points(SegmentAnythingInferenceSettings::new(image).add_goal_point(x, y))
        .unwrap();

    images.save("out.png").unwrap();
}

pub fn segment_everything( &self, image: DynamicImage, ) -> Result<Vec<DynamicImage>, SegmentAnythingInferenceError>

Segment everything in an image. Returns a list of DynamicImage masks.

Example

use segment_anything_rs::*;

let model = SegmentAnything::builder().build().unwrap();
let image = image::open("examples/landscape.jpg").unwrap();
let images = model.segment_everything(image).unwrap();
for (i, img) in images.iter().enumerate() {
    img.save(&format!("{}.png", i)).unwrap();
}

Examples found in repository

examples/segment-everything.rs (line 6)

fn main() {
    let model = SegmentAnything::builder().build().unwrap();
    let image = image::open("examples/landscape.jpg").unwrap();
    let images = model.segment_everything(image).unwrap();
    for (i, img) in images.iter().enumerate() {
        img.save(format!("{}.png", i)).unwrap();
    }
}