add reverse mask

2024-01-30 13:30:43 +08:00
parent e5f71ae535
commit cdac68a9a9
5 changed files with 34 additions and 17 deletions
--- a/iopaint/tests/test_adjust_mask.py
+++ b/iopaint/tests/test_adjust_mask.py
@@ -13,3 +13,5 @@ def test_adjust_mask():
    cv2.imwrite(str(save_dir / "adjust_mask_expand.png"), res_mask)
    res_mask = adjust_mask(mask, 20, "shrink")
    cv2.imwrite(str(save_dir / "adjust_mask_shrink.png"), res_mask)
+    res_mask = adjust_mask(mask, 20, "reverse")
+    cv2.imwrite(str(save_dir / "adjust_mask_reverse.png"), res_mask)