前言

BackgroundRemover是一个命令行工具，用于从视频 和图像中删除背景

1、依赖

Requirements
python <= 3.6
python3.6-dev #or what ever version of python you using
torch and torchvision stable version https://pytorch.org
ffmpeg 4.4+

1
2
3

pip install python3.6-dev
pip install ffmpeg
pip install torch

2、安装

1

pip install backgroundremover

3、下载模型并设置

如果不提前下载模型的话，在首次执行的时候会自动下载，但是需要从 google 网盘下载，因此需要有翻墙能力。
如果手工下载可以参考下面地址：
1、https://github.com/xuebinqin/U-2-Net#usage-for-salient-object-detection，参考章节"Download the u2net_portrait.pth"
2、https://pan.baidu.com/s/1FbUIBJsEua7G1vYo06lSmg 提取码: o0uu
3、https://drive.google.com/u/0/uc?id=1rbSTGKAE-MTxBYHd-51l2hMOQPT_7EPy&export=download

1
2
3
4
5
6
7
8

# Window：
c:/Windows/user/.u2net/u2net.pth

# Mac:
/Users/lyc/.u2net/u2net.pth

# Linux:
/root/.u2net/u2net.pth

4、图片

从本地文件图像中删除背景

1

backgroundremover -i "/path/to/image.jpeg" -o "output.png"

高级用法

有时可以通过打开 alpha matting 来获得更好的结果

1

backgroundremover -i "/path/to/image.jpeg" -a -ae 15 -o "output.png"

切换模型u2netp, u2net, or u2net_human_seg

1

backgroundremover -i "/path/to/image.jpeg" -m "u2net_human_seg" -o "output.png"

5、视频

从视频中删除背景并制作透明mov

1

backgroundremover -i "/path/to/video.mp4" -tv -o "output.mov"

从本地视频中删除背景并将其覆盖在其他视频上

1

backgroundremover -i "/path/to/video.mp4" -tov -tv "/path/to/videtobeoverlayed.mp4" -o "output.mov"

从视频中删除背景并制作透明gif

1

backgroundremover -i "/path/to/video.mp4" -tg -o "output.gif"

制作遮罩键文件（绿屏覆盖）
为制作蒙版文件

1

backgroundremover -i "/path/to/video.mp4" -mk -o "output.matte.mp4"

高级用法

更改视频的帧率（默认设置为 30）

1

backgroundremover -i "/path/to/video.mp4" -fr 30 -tv -o "output.mov"

更改视频的 gpu 批量大小（默认设置为 1）

1

backgroundremover -i "/path/to/video.mp4" -gb 4 -tv -o "output.mov"

更改处理视频的工作人员数量（默认设置为 1）

1

backgroundremover -i "/path/to/video.mp4" -wn 4 -tv -o "output.mov"

切换模型u2netp, u2net, or u2net_human_seg

1

backgroundremover -i "/path/to/video.mp4" -m "u2net_human_seg"-tv -o "output.mov"