nodes/chflame163-ComfyUI_LayerStyle
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

change the RMBG model path to ComfyUI's models/rmbg path

Browse Source
This commit is contained in:
chflame163
2024-02-20 15:16:28 +08:00
parent 4798a13de9
commit 22b91a9782
3 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.MD
+2 -2
View File
@@ -667,8 +667,8 @@ Remove background. compared to the similar background removal nodes, this node h
This node combines the Alpha Matte node of Spacepxl's [ComfyUI-Image-Filters](https://github.com/spacepxl/ComfyUI-Image-Filters) and the functionality of ZHO-ZHO-ZHO's [ComfyUI-BRIA_AI-RMBG](https://github.com/ZHO-ZHO-ZHO/ComfyUI-BRIA_AI-RMBG).
*Download the [BRIA Background Removal v1.4](https://huggingface.co/briaai/RMBG-1.4) model file (model. pth) to the /ComfyUI/models/RMBG-1.4 folder. This model was developed by BRIA AI and can be used as an open-source model for non-commercial purposes.
**After downloading the model, recommended to add a line 'rmbg: models/RMBG-1.4' in ComfyUI/extra_model_paths.yaml. (This step is not mandatory)
*Download the [BRIA Background Removal v1.4](https://huggingface.co/briaai/RMBG-1.4) model file (model. pth) to the /ComfyUI/models/rmbg/RMBG-1.4 folder. This model was developed by BRIA AI and can be used as an open-source model for non-commercial purposes.
**After downloading the model, recommended to add a line 'rmbg: models/rmbg/RMBG-1.4' in ComfyUI/extra_model_paths.yaml. (This step is not mandatory)
![image](image/rembg_ultra_example.png)
README_CN.MD
+2 -2
View File
@@ -654,9 +654,9 @@
去除背景。与类似的背景移除节点相比，这个节点具有超高的边缘细节。
本节点结合了spacepxl的[ComfyUI-Image-Filters](https://github.com/spacepxl/ComfyUI-Image-Filters)的Alpha Matte节点，以及ZHO-ZHO-ZHO的[ComfyUI-BRIA_AI-RMBG](https://github.com/ZHO-ZHO-ZHO/ComfyUI-BRIA_AI-RMBG)的功能。
*将[BRIA Background Removal v1.4](https://huggingface.co/briaai/RMBG-1.4)模型文件(model.pth)下载至/ComfyUI/models/RMBG-1.4文件夹。该模型由 BRIA AI 开发，可作为非商业用途的开源模型。
*将[BRIA Background Removal v1.4](https://huggingface.co/briaai/RMBG-1.4)模型文件(model.pth)下载至/ComfyUI/models/rmbg/RMBG-1.4文件夹。该模型由 BRIA AI 开发，可作为非商业用途的开源模型。
**下载模型后，建议在ComfyUI/extra_model_paths.yaml中添加一行'rmbg: models/RMBG-1.4'。(此步骤非必须）
**下载模型后，建议在ComfyUI/extra_model_paths.yaml中添加一行'rmbg: models/rmbg/RMBG-1.4'。(此步骤非必须）
![image](image/rembg_ultra_example.png)
py/imagefunc.py
+2 -1
View File
@@ -691,7 +691,8 @@ def load_RMBG_model():
except:
pass
if not os.path.exists(model_path):
model_path = os.path.join(folder_paths.models_dir, _path)
model_path = os.path.join(folder_paths.models_dir, "rmbg", _path)
log(model_path)
if not os.path.exists(model_path):
model_path = os.path.join(os.path.dirname(current_directory), _path)