55 lines
1.4 KiB
Markdown
55 lines
1.4 KiB
Markdown
---
|
|
tags:
|
|
- Mask
|
|
- MaskRegion
|
|
---
|
|
|
|
# Masks to Tensors
|
|
## Documentation
|
|
- Class name: `SaltMasksToImages`
|
|
- Category: `SALT/Masking`
|
|
- Output node: `False`
|
|
|
|
This node is designed to convert a collection of mask images into a tensor representation, facilitating the transition from image processing to tensor operations for further computational tasks.
|
|
## Input types
|
|
### Required
|
|
- **`masks`**
|
|
- The masks input represents a collection of mask images that are to be converted into tensor format. This conversion is crucial for enabling subsequent tensor-based operations and analyses.
|
|
- Comfy dtype: `MASK`
|
|
- Python dtype: `List[torch.Tensor]`
|
|
## Output types
|
|
- **`images`**
|
|
- Comfy dtype: `IMAGE`
|
|
- The output is a tensor comprising the converted images from the input masks, ready for further processing or analysis.
|
|
- Python dtype: `torch.Tensor`
|
|
## Usage tips
|
|
- Infra type: `GPU`
|
|
- Common nodes: unknown
|
|
|
|
|
|
## Source code
|
|
```python
|
|
class SaltMasksToImages:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"masks": ("MASK",)
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("IMAGE",)
|
|
RETURN_NAMES = ("images",)
|
|
|
|
CATEGORY = f"{NAME}/Masking"
|
|
FUNCTION = "convert"
|
|
|
|
def convert(self, masks):
|
|
images = []
|
|
for mask in masks:
|
|
images.append(pil2tensor(mask2pil(mask)))
|
|
images = torch.cat(images, dim=0)
|
|
return (images, )
|
|
|
|
```
|