图片去背景

复制from rembg import remove
from PIL import Image
import sys

# 输入输出路径
input_path = sys.argv[1] if len(sys.argv) > 1 else "input.jpg"
output_path = sys.argv[2] if len(sys.argv) > 2 else "output.png"

# 打开图片，移除背景，保存为 PNG（透明背景）
with open(input_path, "rb") as inp:
    input_data = inp.read()
    output_data = remove(input_data)

with open(output_path, "wb") as out:
    out.write(output_data)

print(f"已保存透明背景图片至 {output_path}")

# 使用示例（命令行）：
# python remove_bg.py input.jpg output.png

复制package main

import (
	"fmt"
	"io/ioutil"
	"log"
	"os"

	"github.com/danieldk/go-rembg"
)

func main() {
	inputPath := "input.jpg"
	if len(os.Args) > 1 {
		inputPath = os.Args[1]
	}
	outputPath := "output.png"
	if len(os.Args) > 2 {
		outputPath = os.Args[2]
	}

	// 读取输入图片
	inputBytes, err := ioutil.ReadFile(inputPath)
	if err != nil {
		log.Fatalf("读取输入文件失败: %v", err)
	}

	// 移除背景
	outputBytes, err := rembg.Remove(inputBytes)
	if err != nil {
		log.Fatalf("移除背景失败: %v", err)
	}

	// 写入输出文件
	err = ioutil.WriteFile(outputPath, outputBytes, 0644)
	if err != nil {
		log.Fatalf("写入输出文件失败: %v", err)
	}

	fmt.Printf("已保存透明背景图片至 %s\n", outputPath)
}

// 使用示例（命令行）：
// go run main.go input.jpg output.png

复制const { removeBackground } = require('@imgly/background-removal');
const fs = require('fs');

async function removeBg(inputPath, outputPath) {
  try {
    // 读取输入图片
    const inputBuffer = fs.readFileSync(inputPath);
    
    // 移除背景（返回 Buffer）
    const outputBuffer = await removeBackground(inputBuffer);
    
    // 保存为 PNG
    fs.writeFileSync(outputPath, outputBuffer);
    console.log(`已保存透明背景图片至 ${outputPath}`);
  } catch (error) {
    console.error('移除背景失败:', error);
  }
}

// 使用示例（命令行）：
// node remove_bg.js input.jpg output.png
const inputPath = process.argv[2] || 'input.jpg';
const outputPath = process.argv[3] || 'output.png';
removeBg(inputPath, outputPath);