forked from storyicon/comfyui_segment_anything
-
Notifications
You must be signed in to change notification settings - Fork 1
/
__init__.py
22 lines (18 loc) · 920 Bytes
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
from .node import *
from .install import *
NODE_CLASS_MAPPINGS = {
'SAMModelLoader (segment anything)': SAMModelLoader,
'GroundingDinoModelLoader (segment anything)': GroundingDinoModelLoader,
'GroundingDinoSAMSegment (segment anything)': GroundingDinoSAMSegment,
'GenerateVITMatte (segment anything)': GenerateVITMatte,
'VITMatteTransformersModelLoader (segment anything)': VITMatteTransformersModelLoader,
'MaskToTrimap (segment anything)': MaskToTrimap,
'TrimapToMask (segment anything)': TrimapToMask,
'InvertMask (segment anything)': InvertMask,
"IsMaskEmpty": IsMaskEmptyNode,
"BoundingBox (segment anything)":BoundingBox,
"MaskToBoundingBox (segment anything)":MaskToBoundingBox,
"BoundingBoxSAMSegment (segment anything)":BoundingBoxSAMSegment,
"GroundingDinoBoundingBoxes (segment anything)":GroundingDinoBoundingBoxes
}
__all__ = ['NODE_CLASS_MAPPINGS']