huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 02:58:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

[segment-anything] add multi point logic for demo site (#1002)

Browse Source 
* [segment-anything] add multi point logic for demo site

* [segment-anything] remove libs and update functions
This commit is contained in:
lichin-lin
2023-10-01 18:25:22 +01:00
committed by GitHub
parent 096dee7073
commit 41143db1af
3 changed files with 29 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-wasm-examples/segment-anything/samWorker.js
View File

@ -141,7 +141,7 @@ self.addEventListener("message", async (event) => {
}
self.postMessage({ status: "segmenting", message: "Segmenting" });
const { mask, image } = sam.mask_for_point(points.x, points.y);
const { mask, image } = sam.mask_for_point({ points });
const maskDataURL = await createImageCanvas(mask, image);
// Send the segment back to the main thread as JSON
self.postMessage({