stable-diffusion-webui-rembg是我非常喜欢的sd-webui插件,PS抠图手残党的福音,一键抠图太爽啦。对于主体明确线条简单的图片(如汽车、服饰简洁的人像等)效果相当好;即便对于毛茸茸的对象(如动物)效果差强人意,那也肯定比我手工做的质量好。更重要的是速度还算快,即便没有用到GPU。
插件的安装很简单,在扩展的可下载栏位中搜索并安装即可。但是装完后第一次使用遇到些问题。
首先就是模型的下载。由于国内访问github老抽风,因此可以预先下载好。具体下载地址可以在AKI一键启动包的控制台中找到。我共享了一个通用模型(isnet-general-use)放到了资源里。下载后放入D:\sd-webui-aki-v4.7\.cache\u2net 下(当然,D:\sd-webui-aki-v4.7\需要替换成你实际的)。相应的,移除背景模型里也只能选这个。其它模型针对各自的场景做了优化,但这个通用模型对我是已经够用了。
然后实际运行的时候报错:
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 "D:\sd-webui-aki-v4.7\python\lib\site-packages\onnxruntime\capi\onnxruntime_providers_tensorrt.dll"
对我有效的解决方案(重装onnxruntime):
rem 请改成你实际的python目录
cd D:\sd-webui-aki-v4.7\python
rem 这步可选
python.exe -m pip install --upgrade pip -i https://pypi.tuna.tsinghua.edu.cn/simple
rem 核心处置
python.exe .\Scripts\pip3.exe uninstall onnxruntime-gpu
python.exe .\Scripts\pip3.exe install onnxruntime --user -i https://pypi.tuna.tsinghua.edu.cn/simple
测试效果: