deduplication pass
This commit is contained in:
@@ -3,7 +3,7 @@ import numpy as np
|
||||
from backend.node_registry import register_node
|
||||
from backend.execution_context import emit_overlay
|
||||
from backend.data_types import DataField, datafield_to_uint8, encode_preview
|
||||
from backend.nodes.helpers import _parse_mask_strokes, _rasterize_mask
|
||||
from backend.nodes.helpers import _parse_mask_strokes, _rasterize_mask, bool_to_mask, mask_to_bool
|
||||
|
||||
|
||||
@register_node(display_name="Draw Mask")
|
||||
@@ -37,7 +37,7 @@ class DrawMask:
|
||||
strokes = _parse_mask_strokes(mask_paths)
|
||||
mask = _rasterize_mask(field.xres, field.yres, strokes, pen_size)
|
||||
if invert:
|
||||
mask = np.where(mask > 127, np.uint8(0), np.uint8(255))
|
||||
mask = bool_to_mask(~mask_to_bool(mask))
|
||||
|
||||
emit_overlay({
|
||||
"kind": "mask_paint",
|
||||
|
||||
Reference in New Issue
Block a user