28 lines
691 B
Python
28 lines
691 B
Python
from .nodes import *
|
|
from typing_extensions import override
|
|
|
|
class Sam3Extension(ComfyExtension):
|
|
@override
|
|
async def get_node_list(self) -> list[type[io.ComfyNode]]:
|
|
return [
|
|
LoadSam3Model,
|
|
Sam3ImageSegmentation,
|
|
Sam3VideoSegmentation,
|
|
Sam3VideoModelExtraConfig,
|
|
Sam3Visualization,
|
|
Sam3GetObjectIds,
|
|
Sam3GetObjectMask,
|
|
StringToBBox,
|
|
FramesEditor,
|
|
]
|
|
|
|
async def comfy_entrypoint() -> Sam3Extension:
|
|
return Sam3Extension()
|
|
|
|
# Web directory for custom UI (interactive SAM3 detector)
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
# Export for ComfyUI
|
|
__all__ = ['WEB_DIRECTORY']
|
|
|