xuxinyuancode / rembg-node

Rembg lets you easily remove backgrounds from images using the U2-Net AI model

Rembg Node

Rembg lets you easily remove backgrounds from images using the U2-Net AI model

This is a loose port of the original Rembg for Python, big thanks to @danielgatis

It uses sharp for input and output so you can easily integrate it

The masking algorithm isn't fully complete yet, but the results are pretty good already!

yarn add sharp rembg-node

npm install sharp rembg-node

Example

import { Rembg } from "rembg-node";
import sharp from "sharp";

// const { Rembg } = require("rembg-node");
// const sharp = require("sharp");

(async () => {
	const input = sharp("test-input.jpg");

	// optional arguments
	const rembg = new Rembg({
		logging: true,
	});

	const output = await rembg.remove(input);

	await output.webp().toFile("test-output.webp");

	// optionally you can use .trim() too!
	await output.trim().webp().toFile("test-output-trimmed.webp");
})();