15 lines
403 B
Python
15 lines
403 B
Python
from .node import *
|
|
# from .install import *
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
'SAMModelLoader (segment anything)': SAMModelLoader,
|
|
'GroundingDinoModelLoader (segment anything)': GroundingDinoModelLoader,
|
|
'GroundingDinoSAMSegment (segment anything)': GroundingDinoSAMSegment,
|
|
'InvertMask (segment anything)': InvertMask,
|
|
"IsMaskEmpty": IsMaskEmptyNode,
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS']
|
|
|
|
|