在电商直播源码中，有时候需要去除图片或视频的背景，电商直播源码可以怎样做呢，接下来就让我们一起看一下。

环境

前言

backgroundremover是一款开源免费的命令行工具，专门用来消除图片和视频中的背景，使用 python语言编写。

可以使用 pip安装

pip install backgroundremover

当然，下载源码进行安装也是可以的

git clone https://github.com/nadermx/backgroundremover.git
cd backgroundremover
python setup.py install

目前最新版本为0.1.9，后面测试也是基于这个版本进行。

接下来还需要安装 ffmpeg，它是用来进行电商直播源码的视频处理的。从官网下载编译好的二进制文件，尽量下载较新的版本，并将其路径加入到系统环境变量中

最后安装下 torch，由于本机有 gpu，安装的是 gpu版本的，版本要求1.7以上

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

准备一张带背景的原图

执行下面命令就可以去除背景，操作非常简单

backgroundremover -i original.jpg -o output.png

操作过程中会到 google driver去下载所需要的 u2net模型，但是由于国内的网络环境，无法访问

这时候，可以通过下面的链接去下载

下载后的模型文件，将其放到 ~/.u2net文件夹下，如果文件夹不存在就自己创建

准备就绪后，重新执行上面的去背指令，就可以得到去掉背景的图片了

backgroundremover可以使用三种模型来达到图片视频去背的目的，分别是 u2netp、 u2net和 u2net_human_seg

比如说要使用 u2net_human_seg，可以使用 -m参数

backgroundremover -i original.jpg -m u2net_human_seg -o output.png

如果要处理视频的话，可以这样

backgroundremover -i test.mp4 -tv -o output.mov

视频背景去除后叠加到另一个视频上

backgroundremover -i test.mp4 -tov -tv videtobeoverlayed.mp4 -o output.mov

视频去背后生成 gif动图

backgroundremover -i test.mp4 -tg -o output.gif

以上就是电商直播源码出去图片或视频背景的相关内容了。

声明：本文由云豹科技转发自迷途小书童的Note博客，如有侵权请联系作者删除