Struct segment_anything_rs::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)
                .unwrap()
                .add_goal_point(x, y),
        )
        .unwrap();

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

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

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)
            .unwrap()
            .add_goal_point(x, y),
    )
    .unwrap();

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

Examples found in repository

examples/segment-anything.rs (lines 9-13)

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)
                .unwrap()
                .add_goal_point(x, y),
        )
        .unwrap();

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

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

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();
    }
}