diff options
| author | GeauxEric <yunding.eric@gmail.com> | 2023-09-25 04:14:10 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-25 12:14:10 +0100 |
| commit | 7f2bbcf7468b451c617519c92059957d7a89183e (patch) | |
| tree | 7db867a35babc80cfbe700ec3ac58d83167c80e3 /candle-wasm-examples/segment-anything | |
| parent | dc47224ab9d34c8f4ea0e6ce87d964a030eae89c (diff) | |
| download | candle-7f2bbcf7468b451c617519c92059957d7a89183e.tar.gz candle-7f2bbcf7468b451c617519c92059957d7a89183e.tar.bz2 candle-7f2bbcf7468b451c617519c92059957d7a89183e.zip | |
[segment-anything] Support multi-point as the prompt input (#945)
* [sam] Support multi-point prompts
* [segment-anything] Pass points by reference
* [segment-anything] Update example code and image
* Fix clippy lint.
---------
Co-authored-by: Yun Ding <yunding@nvidia.com>
Co-authored-by: laurent <laurent.mazare@gmail.com>
Diffstat (limited to 'candle-wasm-examples/segment-anything')
| -rw-r--r-- | candle-wasm-examples/segment-anything/src/bin/m.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-wasm-examples/segment-anything/src/bin/m.rs b/candle-wasm-examples/segment-anything/src/bin/m.rs index acd903b0..a56e48c2 100644 --- a/candle-wasm-examples/segment-anything/src/bin/m.rs +++ b/candle-wasm-examples/segment-anything/src/bin/m.rs @@ -94,7 +94,7 @@ impl Model { &embeddings.data, embeddings.height as usize, embeddings.width as usize, - Some((x, y)), + &[(x, y)], false, )?; let iou = iou_predictions.flatten(0, 1)?.to_vec1::<f32>()?[0]; |