FluttyProger commited on
Commit
a52a80e
1 Parent(s): 1c70b30

Delete maskload.py

Browse files
Files changed (1) hide show
  1. 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
- }