Struct segment_anything_rs::SegmentAnythingInferenceSettings

pub struct SegmentAnythingInferenceSettings { /* private fields */ }

Settings for running inference on SegmentAnything.

Implementations

impl SegmentAnythingInferenceSettings

pub fn new<I: GenericImageView<Pixel = Rgba<u8>>>(input: I) -> Result<Self>

Creates a new SegmentAnythingInferenceSettings from an image.

Examples found in repository

examples/segment-anything.rs (line 10)

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 set_threshold(self, threshold: f32) -> Self

Sets the detection threshold for the mask, 0 is the default value.

• A negative values makes the model return a larger mask.
• A positive makes the model return a smaller mask.

pub fn add_goal_point(self, x: impl Into<f64>, y: impl Into<f64>) -> Self

Add a point to the list of points to segment.

Examples found in repository

examples/segment-anything.rs (line 12)

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 set_goal_points(self, points: Vec<(f64, f64)>) -> Self

Set the list of points to segment.

pub fn add_avoid_points(self, x: impl Into<f64>, y: impl Into<f64>) -> Self

Add a point to the list of points to avoid.

pub fn set_avoid_points(self, points: Vec<(f64, f64)>) -> Self

Set the list of points to avoid.

pub fn set_image<I: GenericImageView<Pixel = Rgba<u8>>>( self, image: I ) -> Result<Self>

Set the image to segment.