huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix the multiple points case for sam. (#998)

Browse Source
This commit is contained in:
Laurent Mazare
2023-09-29 22:39:43 +02:00
committed by GitHub
parent 0ac2db577b
commit d188d6a764
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-transformers/src/models/segment_anything/sam.rs
View File

@ -171,8 +171,8 @@ impl Sam {
[x, y]
})
.collect::<Vec<_>>();
let points = Tensor::from_vec(xys, (n_points, 1, 2), img_embeddings.device())?;
let labels = Tensor::ones((n_points, 1), DType::F32, img_embeddings.device())?;
let points = Tensor::from_vec(xys, (1, n_points, 2), img_embeddings.device())?;
let labels = Tensor::ones((1, n_points), DType::F32, img_embeddings.device())?;
Some((points, labels))
};
let points = points.as_ref().map(|(x, y)| (x, y));