zho-zho-zho/comfyui-birefnet-zho

Better version for BiRefNet in ComfyUI | Both img & video

ComfyUI-BiRefNet-ZHO

Better version for BiRefNet in ComfyUI | Both img & video

项目介绍 | Info

• 对 BiRefNet 的非官方实现

• 与 viperyl/ComfyUI-BiRefNet 插件区别：

  • 原版插件：只能简单输出蒙版，不方便用，也不能处理视频

  • 新版插件：

    1）模型加载和图像处理相分离，提升速度（和我之前做的 BRIA RMBG in ComfyUI 插件一致）

    2）可以直接输出透明背景的 PNG 图

    3）可以直接抠视频

• BiRefNet 模型：目前最好的开源可商用背景抠除模型

• 版本：V1.0 同时支持 图像和视频 处理

视频演示

BiRefNet.mp4

安装 | Install

• 所需依赖：timm，如已安装无需运行 requirements.txt，只需 git 项目即可

• 推荐使用管理器 ComfyUI Manager 安装（On the Way）

• 手动安装：

  1. cd custom_nodes
  2. git clone https://github.com/ZHO-ZHO-ZHO/ComfyUI-BiRefNet-ZHO.git
  3. cd custom_nodes/ComfyUI-BiRefNet-ZHO
  4. pip install -r requirements.txt
  5. 重启 ComfyUI

使用说明 | How to Use

• 将 BiRefNet 中的 6 个模型均下载至./models/BiRefNet

• 节点：

  

  • 🧹BiRefNet Model Loader：自动加载 BiRefNet 模型
  • 🧹BiRefNet：去除背景

更新日志

• 20240401

  V1.0 同时支持 图像和视频 处理（支持批量处理）、支持输出 mask 功能

  创建项目

Stars

关于我 | About me

📬 联系我：

• 邮箱：[email protected]
• QQ 群：839821928

🔗 社交媒体：

• 个人页：-Zho-
• Bilibili：我的B站主页
• X（Twitter）：我的Twitter
• 小红书：我的小红书主页

💡 支持我：

• B站：B站充电
• 爱发电：为我充电

Credits

BiRefNet

代码参考了 viperyl/ComfyUI-BiRefNet 感谢！

comfyui-birefnet-zho's Issues

Attempting to deserialize object on a CUDA

File "/Users/shijin/ComfyUI/custom_nodes/ComfyUI-BiRefNet-ZHO/models/backbones/build_backbone.py", line 29, in load_weights save_model = torch.load(config.weights[model_name])

在CPU模式下，这里报错，缺map_location参数

Ability to threshold?

Is it possible to have the ability to have a threshholding option? or is it a 1 shot process no matter what

Missing dependency of opencv-python

requirements.txt should include opencv-python

import failed in MAC

I have installed all the requirements, but it import failed.

I try to uninstall and install again , it did not work.

I use Macbook pro m2 max.

thx

Could not run 'torchvision::deform_conv2d' with arguments from the 'CUDA' backend.

torch-2.2.1+cu121-cp312-cp312-win_amd64，只能等以后版本支持了。
大佬有没有考虑增加一个cpu的选项

请教

您好，请问图和视频去背景的工作流json文件是否可以提供一下，我初次尝试comfyui，加载模型后不知道怎么运行了，下面是我的comfyui界面截图：

期待您的回复

Workflow, please

Could you provide workflow of video masking, thanks.

此插件指定的地址下载的模型，使用时报错

除了标记的两个模型不报错，其他几个全部报错