diff --git a/README.MD b/README.MD index ae938c3..1ce1c57 100644 --- a/README.MD +++ b/README.MD @@ -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) diff --git a/README_CN.MD b/README_CN.MD index 685b9cf..95908db 100644 --- a/README_CN.MD +++ b/README_CN.MD @@ -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) diff --git a/py/imagefunc.py b/py/imagefunc.py index cfdf48b..dd3df8d 100644 --- a/py/imagefunc.py +++ b/py/imagefunc.py @@ -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)