add reverse mask

iopaint/helper.py

@@ -354,10 +354,14 @@ def concat_alpha_channel(rgb_np_img, alpha_channel) -> np.ndarray:
 
 
 def adjust_mask(mask: np.ndarray, kernel_size: int, operate):
+    # fronted brush color "ffcc00bb"
     # kernel_size = kernel_size*2+1
     mask[mask >= 127] = 255
     mask[mask < 127] = 0
-    # fronted brush color "ffcc00bb"
+
+    if operate == "reverse":
+        mask = 255 - mask
+    else:
         kernel = cv2.getStructuringElement(
             cv2.MORPH_ELLIPSE, (2 * kernel_size + 1, 2 * kernel_size + 1)
         )

iopaint/schema.py

@@ -304,12 +304,12 @@ class SwitchModelRequest(BaseModel):
     name: str
 
 
-AdjustMaskOperate = Literal["expand", "shrink"]
+AdjustMaskOperate = Literal["expand", "shrink", "reverse"]
 
 
 class AdjustMaskRequest(BaseModel):
     mask: str = Field(
         ..., description="base64 encoded mask. 255 means area to do inpaint"
     )
-    operate: AdjustMaskOperate = Field(..., description="expand or shrink")
+    operate: AdjustMaskOperate = Field(..., description="expand/shrink/reverse")
     kernel_size: int = Field(5, description="Kernel size for expanding mask")

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)

web_app/src/components/SidePanel/DiffusionOptions.tsx

@@ -839,6 +839,17 @@ const DiffusionOptions = () => {
                   Shrink
                 </div>
               </Button>
+
+              <Button
+                variant="outline"
+                className="p-1 h-8"
+                onClick={() => adjustMask("reverse")}
+                disabled={isProcessing}
+              >
+                <div className="flex items-center gap-1 select-none">
+                  Reverse
+                </div>
+              </Button>
             </div>
 
             <Button

web_app/src/lib/types.ts

@@ -126,4 +126,4 @@ export enum PowerPaintTask {
   outpainting = "outpainting",
 }
 
-export type AdjustMaskOperate = "expand" | "shrink"
+export type AdjustMaskOperate = "expand" | "shrink" | "reverse"