# video-background-removal

**Repository Path**: weyee/video-background-removal

## Basic Information

- **Project Name**: video-background-removal
- **Description**: 去除视频背景的工程
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 1
- **Created**: 2025-07-28
- **Last Updated**: 2025-07-28

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 多模型视频抠图工具

本项目旨在打造一个集成多种先进AI视频抠图模型的工具，让用户可以根据不同的视频内容（如真人、动漫、物体等）选择最合适的引擎，轻松实现高质量的背景移除和替换。

## 项目特点

- **多模型支持**：计划集成多种业界领先的抠图模型。
- **高度灵活**：提供丰富的命令行参数，满足不同场景下的需求。
- **使用简单**：清晰的指令和文档，无需深入了解AI技术即可上手。
- **易于扩展**：良好的项目结构，方便未来添加更多新模型。

## 快速开始

### 1. 环境配置

确保您的电脑上已安装 Python (建议 3.8+ 版本) 和 `ffmpeg`。

### 2. 安装依赖

克隆或下载本项目后，在项目根目录下打开终端，运行以下命令安装所有必需的依赖库：

```bash
pip install -r requirements.txt
```

## 当前已集成模型：Robust Video Matting (RVM)

我们集成的第一个模型是强大的 **Robust Video Matting (RVM)**，它特别擅长处理真人视频，并提供了两种不同精度和速度的预训练模型。

### 使用方法 (`rvm_tool.py`)

您可以通过 `rvm_tool.py` 脚本来使用RVM模型。

**基础命令格式:**
```bash
python rvm_tool.py --input-video <输入文件> --output-file <输出文件> [选项]
```

#### **核心选项:**

- `--model <模型名称>`
  选择要使用的RVM模型。
  - `mobilenetv3` (默认): 速度快，性能均衡，适合大多数真人视频。
  - `resnet50`: 精度更高，处理更慢，适合对质量要求极高的场景或处理动漫等非标准内容。

- `--output-format <格式>`
  选择输出文件的格式。
  - `video` (默认): 输出为标准的 `.mp4` 视频文件。
  - `gif`: 输出为背景透明的 `.gif` 动图。**注意：此模式下将忽略所有背景设置。**

#### **视频输出专用选项 (当 `--output-format` 为 `video` 时):**

- `--background-image <图片路径>`
  指定一张图片作为视频的新背景。

- `--background-color <R,G,B>`
  指定一种纯色作为视频的新背景（默认为黑色 `0,0,0`）。

---

### **使用示例**

1.  **【推荐】处理动漫视频，输出透明背景GIF (使用高精度模型)**
    ```bash
    python rvm_tool.py --input-video "你的动漫视频.mp4" --output-file "输出.gif" --output-format "gif" --model "resnet50"
    ```

2.  **处理真人视频，替换为图片背景**
    ```bash
    python rvm_tool.py --input-video "会议录像.mp4" --output-file "新背景会议.mp4" --output-format "video" --background-image "办公室背景.jpg"
    ```

3.  **快速生成绿幕素材**
    ```bash
    python rvm_tool.py --input-video "演讲视频.mp4" --output-file "绿幕演讲.mp4" --output-format "video" --background-color "0,255,0"
    ```

## 新增功能：动漫人物 & 通用物体抠图 (rembg)

为了更好地处理动漫和其他类型的图像，我们集成了 `rembg` 库，它封装了多种强大的预训练模型，效果出众且使用极为方便。

### 使用方法 (`rembg_tool.py`)

我们创建了一个专门的 `rembg_tool.py` 脚本，用于调用 `rembg` 中的模型，特别适合处理动漫内容。

**基础命令格式:**
```bash
python rembg_tool.py --input-video <输入文件> --output-file <输出GIF> [--model <模型名称>]
```

该脚本专注于生成**背景透明的GIF动图**。

#### **模型选择 (`--model`):**

- `isnet-anime` (默认): **动漫专用模型**。为处理二次元内容量身定制，效果最佳。
- `u2net`: 通用性最强的模型之一，细节处理能力出色，适用于各种物体。
- `u2netp`: `u2net`的轻量版本，速度更快。
- ...以及`rembg`支持的其他模型。

### **使用示例**

1.  **【推荐】处理动漫视频 (使用默认的`isnet-anime`模型)**
    ```bash
    python rembg_tool.py --input-video "你的动漫视频.mp4" --output-file "输出.gif"
    ```

2.  **处理一个包含普通物体的视频 (使用`u2net`模型)**
    ```bash
    python rembg_tool.py --input-video "产品展示.mp4" --output-file "产品动图.gif" --model "u2net"
    ```

## 总结

现在，您可以根据视频内容的类型，选择最合适的工具：
- **真人视频**  ->  `rvm_tool.py`
- **动漫视频**  ->  `rembg_tool.py` (推荐)
- **其他物体**  ->  `rembg_tool.py` (可选用 `u2net` 模型)