switch controlnet in webui

2023-05-13 13:45:27 +08:00
parent 0363472adc
commit 3eef8f4dae
16 changed files with 161 additions and 46 deletions
--- a/lama_cleaner/tests/test_plugins.py
+++ b/lama_cleaner/tests/test_plugins.py
@@ -35,6 +35,7 @@ def _save(img, name):
 def test_remove_bg():
    model = RemoveBG()
    res = model.forward(bgr_img)
+    res = cv2.cvtColor(res, cv2.COLOR_RGBA2BGRA)
    _save(res, "test_remove_bg.png")