www.rembg.com API wrapper for Node.js
A simple, FREE AI background removal tool for Node.js. Currently, this is THE ONLY FREE library available for personal usage from API. Check out our website at www.rembg.com for more details.
Get your FREE API Key from the https://www.rembg.com/api-usage PS: you can still use the API Keys without key but it is very limited
npm i --save @remove-background-ai/rembg.js
inputImage can be one of these:
string | Buffer | { base64: string };
// script.mjs file
import { rembg } from '@remove-background-ai/rembg.js';
import dotenv from 'dotenv';
// Load environment variables from .env file
dotenv.config();
// API_KEY will be loaded from the .env file
const API_KEY = process.env.API_KEY;
// log upload and download progress
const onDownloadProgress = console.log;
const onUploadProgress = console.log;
rembg({
apiKey: API_KEY,
inputImage: './input.png',
onDownloadProgress,
onUploadProgress
}).then(({ outputImagePath, cleanup }) => {
console.log(`✅🎉 background removed and saved under path=${outputImagePath}`);
// if called, it will cleanup (remove from disk) your removed background image
// cleanup();
});
if you wish to return a Base64 instead of temporary URL you can use retrunBase64 parameter:
import { rembg } from '../dist/index.js';
import dotenv from 'dotenv';
// Load environment variables from .env file
dotenv.config();
// API_KEY will be loaded from the .env file
const API_KEY = process.env.API_KEY;
// log upload and download progress
const onDownloadProgress = console.log;
const onUploadProgress = console.log;
rembg({
apiKey: API_KEY,
inputImage: './input.png',
onDownloadProgress,
onUploadProgress,
returnBase64: true
}).then(({ base64Image }) => {
console.log(`✅🎉 background removed ${base64Image}`);
});
Since version 1.1.8, the rembg function can accept an input image as a Buffer or Base64 object. Below is a quick demonstration:
import { rembg } from '../dist/index.js';
import dotenv from 'dotenv';
// Load environment variables from .env file
dotenv.config();
const base64Input = 'data:image/png;base64,/9j/4AAQSkZJRgABAQEASABIAAD/.....etc'
// API_KEY will be loaded from the .env file
const API_KEY = process.env.API_KEY;
// log upload and download progress
const onDownloadProgress = console.log;
const onUploadProgress = console.log;
rembg({
apiKey: API_KEY,
inputImage: {base64: base64Input}, // or simply an imageBuffer
onDownloadProgress,
onUploadProgress,
returnBase64: true
}).then(({ base64Image }) => {
console.log(`✅🎉 background removed ${base64Image}`);
});
The library provides an option to return a mask of the image instead of the processed image. This is controlled by the returnMask
parameter.
When set to true
, the function returns a mask. By default (if omitted), this parameter is set to false
.
rembg({
apiKey: API_KEY,
inputImage: './input.jpg',
onDownloadProgress,
onUploadProgress,
returnMask: true, // <----- Set to true to get the mask of the image
returnBase64: true // Set to true to receive the result as a Base64 string
}).then(({ base64Image }) => {
console.log(`✅🎉 Mask retrieved: ${base64Image}`);
});
Below is the generated mask image using the rembg
function with the mask option enabled:
This image demonstrates the result of the mask generation process. The mask typically highlights the main subject of the image with the background removed or made transparent.
This is very useful to work with Stable Diffusion
for perfect area of inpainting, for example.