FluttyProger
commited on
Commit
•
a52a80e
1
Parent(s):
1c70b30
Delete maskload.py
Browse files- maskload.py +0 -39
maskload.py
DELETED
@@ -1,39 +0,0 @@
|
|
1 |
-
from cloths_segmentation.pre_trained_models import create_model
|
2 |
-
|
3 |
-
class GenMaskModela:
|
4 |
-
|
5 |
-
def __init__(self):
|
6 |
-
pass
|
7 |
-
|
8 |
-
@classmethod
|
9 |
-
def INPUT_TYPES(s):
|
10 |
-
"""
|
11 |
-
Return a dictionary which contains config for all input fields.
|
12 |
-
Some types (string): "MODEL", "VAE", "CLIP", "CONDITIONING", "LATENT", "IMAGE", "INT", "STRING", "FLOAT".
|
13 |
-
Input types "INT", "STRING" or "FLOAT" are special values for fields on the node.
|
14 |
-
The type can be a list for selection.
|
15 |
-
|
16 |
-
Returns: `dict`:
|
17 |
-
- Key input_fields_group (`string`): Can be either required, hidden or optional. A node class must have property `required`
|
18 |
-
- Value input_fields (`dict`): Contains input fields config:
|
19 |
-
* Key field_name (`string`): Name of a entry-point method's argument
|
20 |
-
* Value field_config (`tuple`):
|
21 |
-
+ First value is a string indicate the type of field or a list for selection.
|
22 |
-
+ Secound value is a config for type "INT", "STRING" or "FLOAT".
|
23 |
-
"""
|
24 |
-
return {"required": {}}
|
25 |
-
|
26 |
-
CATEGORY = "image"
|
27 |
-
RETURN_TYPES = ("SEGM_MODEL",)
|
28 |
-
|
29 |
-
FUNCTION = "masks_image"
|
30 |
-
def masks_image(self):
|
31 |
-
modela = create_model("Unet_2020-10-30")
|
32 |
-
modela.eval()
|
33 |
-
|
34 |
-
return (modela,)
|
35 |
-
|
36 |
-
|
37 |
-
NODE_CLASS_MAPPINGS = {
|
38 |
-
"GenMaskModela": GenMaskModela,
|
39 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|