added extend mask

MaskNodes.py

@@ -1002,6 +1002,45 @@ class ChangeChannelCount:
         else: # RGB
             return (tensor2rgb(image),)
 
+class ExtendMask:
+    def __init__(self):
+        pass
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "image": ("IMAGE",),
+              
+            },            
+        }
+
+    RETURN_TYPES = ("IMAGE",)
+    FUNCTION = "extend_mask"
+
+    CATEGORY = "Masquerade Nodes"
+
+    def extend_mask(self, image):
+        mask = tensor2mask(image)
+        morphed_white_pixels = torch.argwhere(mask >0)
+        print(morphed_white_pixels.shape)
+        min_y = min(morphed_white_pixels[:, 1])
+        max_y = max(morphed_white_pixels[:, 1])
+        min_x = min(morphed_white_pixels[:, 2])
+        max_x = max(morphed_white_pixels[:, 2])       
+
+    
+        
+        size = mask.size()
+        image_width = size[2]
+        image_height = size[1]
+        mask = torch.zeros((image_height, image_width))
+        mask[int(max_y):int(image_height), int(min_x):int(max_x)+1] = 1
+        return (mask.unsqueeze(0),)
+
+        
+       
+
 class ConstantMask:
     def __init__(self):
         pass
@@ -1313,6 +1352,7 @@ NODE_CLASS_MAPPINGS = {
     "Create QR Code": CreateQRCodeNode,
     "Convert Color Space": ConvertColorSpace,
     "MasqueradeIncrementer": MaqueradeIncrementerNode,
+    "Extend mask": ExtendMask
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
@@ -1338,4 +1378,5 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "Create QR Code": "Create QR Code",
     "Convert Color Space": "Convert Color Space",
     "MasqueradeIncrementer": "Incrementer",
+    "Extend mask": "Extend mask",
 }