从图像和视频中去除背景
项目描述
背景去除剂
BackgroundRemover 是一个命令行工具，用于从视频 和图像中删除背景，由https://BackgroundRemover.app提供给您，这是一个由nadermx制作的应用程序，由该工具提供支持
要求
- 蟒蛇 <= 3.6
- python3.6-dev #或者你使用的任何版本的python
- 火炬和火炬视觉稳定版（https://pytorch.org）
- ffmpeg 4.4+
如何安装torch和fmpeg
转到https://pytorch.org并向下滚动到INSTALL PYTORCH部分并按照说明进行操作。
例如：
PyTorch Build: Stable (1.7.1)
Your OS: Windows
Package: Pip
Language: Python
CUDA: None
安装 ffmpeg 和 python-dev
sudo apt install ffmpeg python3.6-dev
安装
要安装 backgroundremover，请从 pypi 安装
pip install --upgrade pip
pip install backgroundremover
作为 cli 使用
图片
从本地文件图像中删除背景
backgroundremover -i "/path/to/image.jpeg" -o "output.png"
图像背景去除的高级用法
有时可以通过打开 alpha matting 来获得更好的结果。例子：
backgroundremover -i "/path/to/image.jpeg" a -ae 15 -o "output.png"
u2netp在、u2net或之间更改不同背景去除方法的模型u2net_human_seg
backgroundremover -i "/path/to/image.jpeg" -m "u2net_human_seg" -o "output.png"
视频
从视频中删除背景并制作透明mov
backgroundremover -i "/path/to/video.mp4" -tv -o "output.mov"
从本地视频中删除背景并将其覆盖在其他视频上
backgroundremover -i "/path/to/video.mp4" -tov -tv "/path/to/videtobeoverlayed.mp4" -o "output.mov"
从视频中删除背景并制作透明gif
backgroundremover -i "/path/to/video.mp4" -tg -o "output.gif"
制作遮罩键文件（绿屏覆盖）
为总理制作蒙版文件
backgroundremover -i "/path/to/video.mp4" -mk -o "output.matte.mp4"
视频的高级使用
更改视频的帧率（默认设置为 30）
backgroundremover -i "/path/to/video.mp4" -fr 30 -tv -o "output.mov"
更改视频的 gpu 批量大小（默认设置为 1）
backgroundremover -i "/path/to/video.mp4" -gp 4 -tv -o "output.mov"
更改处理视频的工作人员数量（默认设置为 1）
backgroundremover -i "/path/to/video.mp4" -wn 4 -tv -o "output.mov"
u2netp在、u2net或之间更改不同背景去除方法的模型u2net_human_seg
backgroundremover -i "/path/to/video.mp4" -m "u2net_human_seg"-tv -o "output.mov"
去做
- 将逻辑从视频转换为图像，以在图像删除时利用更多 GPU
- 更多地清理文档
- 添加调整并向数据集提供反馈图像或视频的能力
- 其他
拉取请求
公认
如果你喜欢这个图书馆
给我们的项目BackgroundRemover.app或这个 git 的链接，告诉人们你喜欢它或使用它。
项目原因
我们在合并其他部分后制作了自己的包，通过将部分发布为超级用户的悬赏问题等添加了我们自己的一些功能。以及早些时候在hackernews上要求开源图像部分，所以决定添加在视频中，还有更多。
参考
- https://arxiv.org/pdf/2005.09007.pdf
- https://github.com/NathanUA/U-2-Net
- https://github.com/pymatting/pymatting
- https://github.com/danielgatis/rembg
- https://github.com/ecsplendid/rembg-greenscreen
- https://superuser.com/questions/1647590/have-ffmpeg-merge-a-matte-key-file-over-the-normal-video-file-removing-the-backg
- https://superuser.com/questions/1648680/ffmpeg-alphamerge-two-videos-into-a-gif-with-transparent-background/1649339?noredirect=1#comment2522687_1649339
- https://superuser.com/questions/1649817/ffmpeg-overlay-a-video-after-alphamerging-two-others/1649856#1649856
执照
- 版权所有 (c) 2021 年至今Johnathan Nader
- 版权所有 (c) 2020 年至今Lucas Nestler
- 版权所有 (c) 2020 年至今Tim Scarfe 博士
- 版权所有 (c) 2020 年至今Daniel Gatis
获得麻省理工学院许可
