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/schema.py
+++ b/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")