Generate masks for objects in images for tasks like image segmentation, medical imaging, and computer vision applications.

Quickstart

Generate a Mask for an Image

Send an image to a model to generate object masks.

import Bytez from "bytez.js";
import { writeFileSync } from "fs";

const client = new Bytez("YOUR_BYTEZ_KEY_HERE");
const model = client.model("facebook/sam-vit-base");

const inputImage = "https://huggingface.co/datasets/huggingfacejs/tasks/resolve/main/mask-generation/mask-generation-input.png";

const { error, output } = await model.run(inputImage);

if (error) {
  console.error("Error:", error);
} else {
  const { masks, scores } = output;
  console.log({ masks, scores });
  writeFileSync("masks.json", JSON.stringify(masks, null, 2));
}

Demo