minisam
Version:
Plug and play ONNX SAM segmentation in the browser
19 lines (18 loc) • 639 B
TypeScript
import { Tensor } from "onnxruntime-web";
export declare function buildSamFeeds(params: {
clicks: Array<{
x: number;
y: number;
clickType: 0 | 1;
}>;
embedding: Tensor;
originalWidth: number;
originalHeight: number;
}): {
image_embeddings: Tensor;
point_coords: import("onnxruntime-web").TypedTensor<"float32">;
point_labels: import("onnxruntime-web").TypedTensor<"float32">;
orig_im_size: import("onnxruntime-web").TypedTensor<"float32">;
mask_input: import("onnxruntime-web").TypedTensor<"float32">;
has_mask_input: import("onnxruntime-web").TypedTensor<"float32">;
};