18 lines
716 B
Python
18 lines
716 B
Python
from .node import SAM2ModelLoader, GroundingDinoModelLoader, GroundingDinoSAM2Segment, InvertMask, IsMaskEmptyNode
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
'SAM2ModelLoader (segment anything2)': SAM2ModelLoader,
|
|
'GroundingDinoModelLoader (segment anything2)': GroundingDinoModelLoader,
|
|
'GroundingDinoSAM2Segment (segment anything2)': GroundingDinoSAM2Segment,
|
|
'InvertMask (segment anything)': InvertMask,
|
|
"IsMaskEmpty": IsMaskEmptyNode,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
'SAM2ModelLoader': 'SAM2 Model Loader',
|
|
'GroundingDinoModelLoader': 'Grounding Dino Model Loader',
|
|
'GroundingDinoSAM2Segment': 'Grounding Dino SAM2 Segment',
|
|
'InvertMask': 'Invert Mask',
|
|
"IsMaskEmpty": "Is Mask Empty",
|
|
}
|