GithubHelp home page GithubHelp logo

stable-diffusion-webui-rembg's Introduction

Rembg

Extension for webui. Removes backgrounds from pictures.

Find the UI for rembg in the Extras tab after installing the extension.

Installation

Install from webui's Extensions tab.

Credits

stable-diffusion-webui-rembg's People

Contributors

automatic1111 avatar catboxanon avatar ccfco avatar h3rmit-git avatar huotarih avatar richarddally avatar thireus avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

stable-diffusion-webui-rembg's Issues

File extension override

Because transparency is not supported with jpg, if a user has jpg as the save extension in webui settings, it should be ignored or overriden by a separate selection inside the extras tab. Additionally, choosing webp should support transparency and currently webp will just render black on transparency.

dll load failed while importing "onnxruntime_pybind11_state"

Since I installed stable-diffusion-webui-rembg I have this error :

venv "D:\apps\stable-diffusion\webui__automatic1111\webui\venv\Scripts\Python.ex
e"
Python 3.10.6 (main, Sep 19 2022, 15:45:37) [MSC v.1933 64 bit (AMD64)]
Version: v1.7.0
Commit hash: cf2772fab0af5573da775e7437e6acdca424f26e
Launching Web UI with arguments: --xformers --skip-torch-cuda-test --medvram-sdx
l
Style database not found: D:\apps\stable-diffusion\webui__automatic1111\webui\st
yles.csv
ControlNet preprocessor location: D:\apps\stable-diffusion\webui__automatic1111
webui\extensions\sd-webui-controlnet\annotator\downloads
2024-02-21 22:46:36,993 - ControlNet - INFO - ControlNet v1.1.438
2024-02-21 22:46:37,165 - ControlNet - INFO - ControlNet v1.1.438
*** Error loading script: api.py
Traceback (most recent call last):
File "D:\apps\stable-diffusion\webui__automatic1111\webui\modules\scripts.
py", line 469, in load_scripts
script_module = script_loading.load_module(scriptfile.path)
File "D:\apps\stable-diffusion\webui__automatic1111\webui\modules\script_l
oading.py", line 10, in load_module
module_spec.loader.exec_module(module)
File "", line 883, in exec_module
File "", line 241, in call_with_frames_remov
ed
File "D:\apps\stable-diffusion\webui__automatic1111\webui\extensions\stabl
e-diffusion-webui-rembg\scripts\api.py", line 7, in
import rembg
File "D:\apps\stable-diffusion\webui__automatic1111\webui\venv\lib\site-pa
ckages\rembg_init
.py", line 5, in
from .bg import remove
File "D:\apps\stable-diffusion\webui__automatic1111\webui\venv\lib\site-pa
ckages\rembg\bg.py", line 21, in
from .session_factory import new_session
File "D:\apps\stable-diffusion\webui__automatic1111\webui\venv\lib\site-pa
ckages\rembg\session_factory.py", line 4, in
import onnxruntime as ort
File "D:\apps\stable-diffusion\webui__automatic1111\webui\venv\lib\site-pa
ckages\onnxruntime_init_.py", line 56, in
raise import_capi_exception
File "D:\apps\stable-diffusion\webui__automatic1111\webui\venv\lib\site-pa
ckages\onnxruntime_init_.py", line 23, in
from onnxruntime.capi._pybind_state import ExecutionMode # noqa: F401
File "D:\apps\stable-diffusion\webui__automatic1111\webui\venv\lib\site-pa
ckages\onnxruntime\capi_pybind_state.py", line 32, in
from .onnxruntime_pybind11_state import * # noqa
ImportError: DLL load failed while importing onnxruntime_pybind11_state: La
procédure spécifiée est introuvable.


*** Error loading script: postprocessing_rembg.py
Traceback (most recent call last):
File "D:\apps\stable-diffusion\webui__automatic1111\webui\modules\scripts.
py", line 469, in load_scripts
script_module = script_loading.load_module(scriptfile.path)
File "D:\apps\stable-diffusion\webui__automatic1111\webui\modules\script_l
oading.py", line 10, in load_module
module_spec.loader.exec_module(module)
File "", line 883, in exec_module
File "", line 241, in call_with_frames_remov
ed
File "D:\apps\stable-diffusion\webui__automatic1111\webui\extensions\stabl
e-diffusion-webui-rembg\scripts\postprocessing_rembg.py", line 6, in
import rembg
File "D:\apps\stable-diffusion\webui__automatic1111\webui\venv\lib\site-pa
ckages\rembg_init
.py", line 5, in
from .bg import remove
File "D:\apps\stable-diffusion\webui__automatic1111\webui\venv\lib\site-pa
ckages\rembg\bg.py", line 21, in
from .session_factory import new_session
File "D:\apps\stable-diffusion\webui__automatic1111\webui\venv\lib\site-pa
ckages\rembg\session_factory.py", line 4, in
import onnxruntime as ort
File "D:\apps\stable-diffusion\webui__automatic1111\webui\venv\lib\site-pa
ckages\onnxruntime_init_.py", line 56, in
raise import_capi_exception
File "D:\apps\stable-diffusion\webui__automatic1111\webui\venv\lib\site-pa
ckages\onnxruntime_init_.py", line 23, in
from onnxruntime.capi._pybind_state import ExecutionMode # noqa: F401
File "D:\apps\stable-diffusion\webui__automatic1111\webui\venv\lib\site-pa
ckages\onnxruntime\capi_pybind_state.py", line 32, in
from .onnxruntime_pybind11_state import * # noqa
ImportError: DLL load failed while importing onnxruntime_pybind11_state: La
procédure spécifiée est introuvable.

I installed last VC_redist.x64.exe.
I installed : pip install onnxruntime==1.15.0
but I still have this error :(

Please help

when API mode, always remove background.

when use Upscaler with API mode(ex. /sdapi/v1/extra-single-image), it always remove background.

  1. bug & reason
    </scripts/postprocessing_rembg.py >

def process(self, pp: scripts_postprocessing.PostprocessedImage, model, return_mask, alpha_matting, alpha_matting_foreground_threshold, alpha_matting_background_threshold, alpha_matting_erode_size):
if model == "None":
return

input model value is None (no quotation marks)

  1. Solution
    def process(self, pp: scripts_postprocessing.PostprocessedImage, model, return_mask, alpha_matting, alpha_matting_foreground_threshold, alpha_matting_background_threshold, alpha_matting_erode_size):
    if (model == "None" or model == 'None' or model == None):
    return

Very slow since onnxruntime update + CUDA CUDA_MODULE_LOADING Environmental Variable not working

Since onnxruntime updated, I was forced to update tensorrt and cuDNN to get rembg working again.

Unfortunately, it's incredibly slow (1:30) now and I am not able to get it faster again, using a RTX 3080 10GB.

During processing, the terminal is spammed by the following:
[W:onnxruntime:Default, tensorrt_execution_provider.h:63 onnxruntime::TensorrtLogger::log] [2023-03-23 15:28:50 WARNING] CUDA lazy loading is not enabled. Enabling it can significantly reduce device memory usage and speed up TensorRT initialization. See "Lazy Loading" section of CUDA documentation https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#lazy-loading

So, I've tried to set the env variable following AUTOMATIC1111/stable-diffusion-webui#8459 to my web-user.bat.

Unfortunately, neither
set CUDA_MODULE_LOADING="LAZY"
nor
set CUDA_MODULE_LOADING=LAZY

Seem to set the variable.

I'd appreciate help or a information about which version combination of CUDA, cuDNN and tensorrt would get me back to the previous speed.

My current versions:
CUDA 12.1.0
cuDNN 8.8.1.3
tensorrt 8.6.0.12

Thanks in advance!

Bug since recent update

Hi,

I used this extention succesfully a couple of weeks ago. The only thing that changed on my system is that I updated the extentions within stable diffusion. No other update or change has been done since the last time I used this extension.

I found some similar people with the same issue, but nothing seemed to solve it.

Can this be fixed? Or can I download an older version of this extension somewhere.

Windows 10, AMD videocard.

See error below:

2024-02-25 19:25:19.1724639 [E:onnxruntime:Default, provider_bridge_ort.cc:1532 onnxruntime::TryGetProviderInfo_TensorRT] D:\a_work\1\s\onnxruntime\core\session\provider_bridge_ort.cc:1209 onnxruntime::ProviderLibrary::Get [ONNXRuntimeError] : 1 : FAIL : LoadLibrary failed with error 126 "" when trying to load "E:\user\stable-diffusion-webui\venv\lib\site-packages\onnxruntime\capi\onnxruntime_providers_tensorrt.dll"

*************** EP Error ***************
EP Error D:\a_work\1\s\onnxruntime\python\onnxruntime_pybind_state.cc:456 onnxruntime::python::RegisterTensorRTPluginsAsCustomOps Please install TensorRT libraries as mentioned in the GPU requirements page, make sure they're in the PATH or LD_LIBRARY_PATH, and that your GPU is supported.
when using ['TensorrtExecutionProvider', 'CUDAExecutionProvider', 'CPUExecutionProvider']
Falling back to ['CUDAExecutionProvider', 'CPUExecutionProvider'] and retrying.


*** Error completing request
*** Arguments: ('task(mmex2la312idy3j)', 0, <PIL.Image.Image image mode=RGB size=576x720 at 0x2D382187FA0>, None, '', '', True, 0, 1, 512, 512, True, 'None', 'None', 0, False, 1, False, 1, 0, False, 0.9, 0.15, 0.5, False, False, 384, 768, 4096, 409600, 'Maximize area', 0.1, False, ['Deepbooru'], False, ['Horizontal'], False, 0.5, 0.2, None, False, '0', '0', 'inswapper_128.onnx', 'CodeFormer', 1, True, 'None', 1, 1, 1, 0, 0, 0.5, 'CPU', False, 0, 'None', '', None, False, False, 0.5, 0, True, 'u2net_human_seg', False, False, 240, 10, 10) {}
Traceback (most recent call last):
File "E:\user\stable-diffusion-webui\venv\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 419, in init
self._create_inference_session(providers, provider_options, disabled_optimizers)
File "E:\user\stable-diffusion-webui\venv\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 469, in _create_inference_session
self._register_ep_custom_ops(session_options, providers, provider_options, available_providers)
File "E:\user\stable-diffusion-webui\venv\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 516, in _register_ep_custom_ops
C.register_tensorrt_plugins_as_custom_ops(session_options, provider_options[i])
RuntimeError: D:\a_work\1\s\onnxruntime\python\onnxruntime_pybind_state.cc:456 onnxruntime::python::RegisterTensorRTPluginsAsCustomOps Please install TensorRT libraries as mentioned in the GPU requirements page, make sure they're in the PATH or LD_LIBRARY_PATH, and that your GPU is supported.

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "E:\user\stable-diffusion-webui\modules\call_queue.py", line 57, in f
    res = list(func(*args, **kwargs))
  File "E:\user\stable-diffusion-webui\modules\call_queue.py", line 36, in f
    res = func(*args, **kwargs)
  File "E:\user\stable-diffusion-webui\modules\postprocessing.py", line 132, in run_postprocessing_webui
    return run_postprocessing(*args, **kwargs)
  File "E:\user\stable-diffusion-webui\modules\postprocessing.py", line 73, in run_postprocessing
    scripts.scripts_postproc.run(initial_pp, args)
  File "E:\user\stable-diffusion-webui\modules\scripts_postprocessing.py", line 196, in run
    script.process(single_image, **process_args)
  File "E:\user\stable-diffusion-webui\extensions\stable-diffusion-webui-rembg\scripts\postprocessing_rembg.py", line 66, in process
    session=rembg.new_session(model),
  File "E:\user\stable-diffusion-webui\venv\lib\site-packages\rembg\session_factory.py", line 26, in new_session
    return session_class(model_name, sess_opts, providers, *args, **kwargs)
  File "E:\user\stable-diffusion-webui\venv\lib\site-packages\rembg\sessions\base.py", line 31, in __init__
    self.inner_session = ort.InferenceSession(
  File "E:\user\stable-diffusion-webui\venv\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 432, in __init__
    raise fallback_error from e
  File "E:\user\stable-diffusion-webui\venv\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 427, in __init__
    self._create_inference_session(self._fallback_providers, None)
  File "E:\user\stable-diffusion-webui\venv\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 483, in _create_inference_session
    sess.initialize_session(providers, provider_options, disabled_optimizers)
RuntimeError: D:\a\_work\1\s\onnxruntime\core\providers\cuda\cuda_call.cc:121 onnxruntime::CudaCall D:\a\_work\1\s\onnxruntime\core\providers\cuda\cuda_call.cc:114 onnxruntime::CudaCall CUDA failure 35: CUDA driver version is insufficient for CUDA runtime version ; GPU=744336320 ; hostname=DESKTOP-0G6VIHV ; file=D:\a\_work\1\s\onnxruntime\core\providers\cuda\cuda_execution_provider.cc ; line=245 ; expr=cudaSetDevice(info_.device_id);

The downloaded onnx for the Remove Background is saved in c:\Users\USERNAME\.u2net\*

Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net.onnx' to file 'C:\Users\USERNAME\.u2net\u2net.onnx'.
100%|###############################################| 176M/176M [00:00<?, ?B/s]
Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2netp.onnx' to file 'C:\Users\USERNAME\.u2net\u2netp.onnx'.
100%|#############################################| 4.57M/4.57M [00:00<?, ?B/s]
Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net_human_seg.onnx' to file 'C:\Users\USERNAME\.u2net\u2net_human_seg.onnx'.

I believe these should be saved somewhere in the web-ui folders

Alpha matting dont work with Batch from directiory

when I use single image and batch mode when enabled Alpha matting
I get good results, example in screenshots 1 and 2
Снимок экрана 2023-09-14 001026
Снимок экрана 2023-09-14 001351

But when, I use batch from directory
For some reason I get unsatisfactory results, as if the mode Alpha matting is not turned on. Example in screenshots 3
Снимок экрана 2023-09-14 001002

Please update

Please update this so it still works. Installing doesn't do anything (it installs, but theres no extra options, or errors)

LoadLibrary failed with error 126

Error completing request
Arguments: (0, <PIL.Image.Image image mode=RGB size=1024x1536 at 0x250236FB370>, None, '', '', True, 0, 1, 512, 512, True, 'None', 'None', 0, 0, 0, 0, 'u2net', False, False, 240, 10, 10) {}
Traceback (most recent call last):
File "E:\NEW_SD\stable-diffusion-webui\modules\call_queue.py", line 56, in f
res = list(func(*args, **kwargs))
File "E:\NEW_SD\stable-diffusion-webui\modules\call_queue.py", line 37, in f
res = func(*args, **kwargs)
File "E:\NEW_SD\stable-diffusion-webui\modules\postprocessing.py", line 56, in run_postprocessing
scripts.scripts_postproc.run(pp, args)
File "E:\NEW_SD\stable-diffusion-webui\modules\scripts_postprocessing.py", line 130, in run
script.process(pp, **process_args)
File "E:\NEW_SD\stable-diffusion-webui\extensions\stable-diffusion-webui-rembg\scripts\postprocessing_rembg.py", line 53, in process
session=rembg.new_session(model),
File "E:\NEW_SD\stable-diffusion-webui\venv\lib\site-packages\rembg\session_factory.py", line 66, in new_session
ort.InferenceSession(
File "E:\NEW_SD\stable-diffusion-webui\venv\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 360, in init
self._create_inference_session(providers, provider_options, disabled_optimizers)
File "E:\NEW_SD\stable-diffusion-webui\venv\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 408, in _create_inference_session
sess.initialize_session(providers, provider_options, disabled_optimizers)
RuntimeError: D:\a_work\1\s\onnxruntime\core\session\provider_bridge_ort.cc:1106 onnxruntime::ProviderLibrary::Get [ONNXRuntimeError] : 1 : FAIL : LoadLibrary failed with error 126 "" when trying to load "E:\NEW_SD\stable-diffusion-webui\venv\lib\site-packages\onnxruntime\capi\onnxruntime_providers_tensorrt.dll"

After I downloaded the onnx i met this

What is the API for sd webui?

I want to use the API on webui server with the payload as :
payload = {
"image": encode_pil_to_base64(image),
"model": "u2net",
"return_mask": False,
"alpha_matting": False,
"alpha_matting_foreground_threshold": 240,
"alpha_matting_background_threshold": 10,
"alpha_matting_erode_size": 10,
}
but it dos not work.

Is there any docs for remby API?

Tensor.dll error

I got this error:

Error completing request
Arguments: (0, <PIL.Image.Image image mode=RGB size=720x480 at 0x1E6DB78EF50>, None, '', '', True, 0, 2, 512, 512, True, 'None', 'None', 0, 0, 0, 0, 'u2net_human_seg', False, False, 240, 10, 10) {}
Traceback (most recent call last):
File "C:\Users\Amir\stable-diffusion-webui\modules\call_queue.py", line 56, in f
res = list(func(*args, **kwargs))
File "C:\Users\Amir\stable-diffusion-webui\modules\call_queue.py", line 37, in f
res = func(*args, **kwargs)
File "C:\Users\Amir\stable-diffusion-webui\modules\postprocessing.py", line 56, in run_postprocessing
scripts.scripts_postproc.run(pp, args)
File "C:\Users\Amir\stable-diffusion-webui\modules\scripts_postprocessing.py", line 130, in run
script.process(pp, **process_args)
File "C:\Users\Amir\stable-diffusion-webui\extensions\stable-diffusion-webui-rembg\scripts\postprocessing_rembg.py", line 53, in process
session=rembg.new_session(model),
File "C:\Users\Amir\stable-diffusion-webui\venv\lib\site-packages\rembg\session_factory.py", line 66, in new_session
ort.InferenceSession(
File "C:\Users\Amir\stable-diffusion-webui\venv\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 360, in init
self._create_inference_session(providers, provider_options, disabled_optimizers)
File "C:\Users\Amir\stable-diffusion-webui\venv\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 408, in _create_inference_session
sess.initialize_session(providers, provider_options, disabled_optimizers)
RuntimeError: D:\a_work\1\s\onnxruntime\core\session\provider_bridge_ort.cc:1106 onnxruntime::ProviderLibrary::Get [ONNXRuntimeError] : 1 : FAIL : LoadLibrary failed with error 126 "" when trying to load "C:\Users\Amir\stable-diffusion-webui\venv\lib\site-packages\onnxruntime\capi\onnxruntime_providers_tensorrt.dll"

missing feature

save with transparent or another color. We can save with black color background but that require to change the save extension in settings

Hello guys, there some issue with onnxruntime, please help me

venv "C:\stable-diffusion-webui-directml\venv\Scripts\Python.exe"
NVIDIA driver was found.
fatal: No names found, cannot describe anything.
Python 3.10.6 (tags/v3.10.6:9c7b4bd, Aug 1 2022, 21:53:49) [MSC v.1932 64 bit (AMD64)]
Version: 1.7.0
Commit hash: 601f7e3704707d09ca88241e663a763a2493b11a
no module 'xformers'. Processing without...
no module 'xformers'. Processing without...
No module 'xformers'. Proceeding without it.
C:\stable-diffusion-webui-directml\venv\lib\site-packages\pytorch_lightning\utilities\distributed.py:258: LightningDeprecationWarning: pytorch_lightning.utilities.distributed.rank_zero_only has been deprecated in v1.8.1 and will be removed in v2.0.0. You can import it from pytorch_lightning.utilities instead.
rank_zero_deprecation(
Installing onnxruntime-gpu
Traceback (most recent call last):
File "C:\stable-diffusion-webui-directml\launch.py", line 48, in
main()
File "C:\stable-diffusion-webui-directml\launch.py", line 39, in main
prepare_environment()
File "C:\stable-diffusion-webui-directml\modules\launch_utils.py", line 636, in prepare_environment
run_pip("install onnxruntime-gpu", "onnxruntime-gpu")
File "C:\stable-diffusion-webui-directml\modules\launch_utils.py", line 150, in run_pip
return run(
File "C:\stable-diffusion-webui-directml\modules\launch_utils.py", line 122, in run
raise RuntimeError("\n".join(error_bits))
RuntimeError: Couldn't install onnxruntime-gpu.
Command: "C:\stable-diffusion-webui-directml\venv\Scripts\python.exe" -m pip install onnxruntime-gpu --prefer-binary
Error code: 1
stdout: Collecting onnxruntime-gpu
Using cached onnxruntime_gpu-1.17.0-cp310-cp310-win_amd64.whl.metadata (4.3 kB)
Requirement already satisfied: coloredlogs in c:\stable-diffusion-webui-directml\venv\lib\site-packages (from onnxruntime-gpu) (15.0.1)
Requirement already satisfied: flatbuffers in c:\stable-diffusion-webui-directml\venv\lib\site-packages (from onnxruntime-gpu) (23.5.26)
Requirement already satisfied: numpy>=1.21.6 in c:\stable-diffusion-webui-directml\venv\lib\site-packages (from onnxruntime-gpu) (1.23.5)
Requirement already satisfied: packaging in c:\stable-diffusion-webui-directml\venv\lib\site-packages (from onnxruntime-gpu) (23.2)
Requirement already satisfied: protobuf in c:\stable-diffusion-webui-directml\venv\lib\site-packages (from onnxruntime-gpu) (3.20.3)
Requirement already satisfied: sympy in c:\stable-diffusion-webui-directml\venv\lib\site-packages (from onnxruntime-gpu) (1.12)
Requirement already satisfied: humanfriendly>=9.1 in c:\stable-diffusion-webui-directml\venv\lib\site-packages (from coloredlogs->onnxruntime-gpu) (10.0)
Requirement already satisfied: mpmath>=0.19 in c:\stable-diffusion-webui-directml\venv\lib\site-packages (from sympy->onnxruntime-gpu) (1.3.0)
Requirement already satisfied: pyreadline3 in c:\stable-diffusion-webui-directml\venv\lib\site-packages (from humanfriendly>=9.1->coloredlogs->onnxruntime-gpu) (3.4.1)
Using cached onnxruntime_gpu-1.17.0-cp310-cp310-win_amd64.whl (148.6 MB)
Installing collected packages: onnxruntime-gpu

stderr: ERROR: Could not install packages due to an OSError: [WinError 5] : 'C:\stable-diffusion-webui-directml\venv\Lib\site-packages\onnxruntime\capi\onnxruntime_providers_shared.dll'
Check the permissions


i already given the permissions to all users, and it didn't work

Onnxruntime error

Im getting this error everytime i start A1111 but i also dont notice anything technically broken when using the AI. i havent done anything to change any links in the PATH settings. All i could think of is a recent issue with faceswaplabs which i believe also uses onnxruntime but that plugin is working.
Screenshot 2024-03-02 224244

API - providing parameters via the API

I noticed when this extension installed, the background is automatically removed when using the /sdapi/v1/extra-single-image endpoint.

Is there currently a way to disable this and provide parameters, e.g. the preferred model?

Not Working

I installed the extension, when pressing the button it looks like it's working for about 2-3 seconds, displays the image on the right window but no results. I tried changing the settings, again nothing. I doesn't look like it's installed any of the databases, I searched for them and they are nowhere to be found, Running Windows 11.

ModuleNotFoundError: No module named 'rembg'

I have just discovered this script. And after installing it on the fresh web-ui and enabling it in the Extensions tab I got this error

Restarting UI...
Error loading script: postprocessing_rembg.py
Traceback (most recent call last):
  File "C:\stable-diffusion-webui\modules\scripts.py", line 256, in load_scripts
    script_module = script_loading.load_module(scriptfile.path)
  File "C:\stable-diffusion-webui\modules\script_loading.py", line 11, in load_module
    module_spec.loader.exec_module(module)
  File "<frozen importlib._bootstrap_external>", line 883, in exec_module
  File "<frozen importlib._bootstrap>", line 241, in _call_with_frames_removed
  File "C:\stable-diffusion-webui\extensions\stable-diffusion-webui-rembg\scripts\postprocessing_rembg.py", line 5, in <module>
    import rembg
ModuleNotFoundError: No module named 'rembg'

ONNX error

when I use plugin rembg,there is an error message and then console breakdown
1686899254714

Hello, the following problems occur when your plug-in is used, how to solve them?

Hello, the following problems occur when your plug-in is used, how to solve them?

ConnectionError: HTTPSConnectionPool(host='github.com', port=443): Max retries exceeded with url: /danielgatis/rembg/releases/download/v0.0.0/u2netp.onnx (Caused by NewConnectionError('<urllib3.connecti
Snipaste_2023-07-03_20-03-14
on.HTTPSConnection object at 0x00000214642D41C0>: Failed to establish a new connection: [WinError 10061] 由于目标计算机积极拒绝,无法连接。'))
Time taken: 2.25sTorch active/reserved: 2076/2086 MiB, Sys VRAM: 3691/24564 MiB (15.03%)

error 126

I believe you guys have met this dude before -
" RuntimeError: D:\a_work\1\s\onnxruntime\core\session\provider_bridge_ort.cc:1106 onnxruntime::ProviderLibrary::Get [ONNXRuntimeError] : 1 : FAIL : LoadLibrary failed with error 126 "" when trying to load "C:\Users\InsertFakeName\Documents\A1111 Web UI Autoinstaller\stable-diffusion-webui\venv\lib\site-packages\onnxruntime\capi\onnxruntime_providers_tensorrt.dll" "

Ill be honest I'm but a simple man and I dont know shit about fuck. I use a easy install UI and am not even for sure how to pull the command prompt up by itself.

plz help

Cloth_Seg Outputs Concatenated into Single Image

First of all, thanks for providing this! The postprocessing of rembg outputs so far seems better than for other rembg plugins I've tried.

I just wanted to note that currently u2net_cloth_seg appends all of its outputs into one image that's 3x taller than the input.

Also, downloaded models on Linux (Mint) are getting stored in ~/.local/share/.u2net/*.onnx, not in the typical ${application directory}/models/**/*.onnx

Seeing as this commit is 2 hours old, these are probably just things you're still working on. If so, feel free to just close this without further comment! :)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.