add briaai/RMBG-2.0

https://github.com/Sanster/IOPaint/issues/598
2024-11-23 14:56:14 +08:00
parent ecbb73aaae
commit 175c597df8
5 changed files with 53 additions and 1 deletions
--- a/iopaint/tests/test_plugins.py
+++ b/iopaint/tests/test_plugins.py
@@ -33,6 +33,7 @@ person_rgb_img = cv2.resize(person_rgb_img, (512, 512))


 def _save(img, name):
+    name = name.replace("/", "_")
    cv2.imwrite(str(save_dir / name), img)