12 lines
397 B
Python
12 lines
397 B
Python
from .nodes.image_nodes import *
|
|
from .nodes.schedule_nodes import *
|
|
from comfy_api.latest import ComfyExtension, io
|
|
|
|
class BboxNodesExtension(ComfyExtension):
|
|
async def get_node_list(self) -> list[type[io.ComfyNode]]:
|
|
return [BboxesVisualize, TrackingVisualize, TrackingToBBoxScheduler, BBoxScheduler]
|
|
|
|
async def comfy_entrypoint() -> ComfyExtension:
|
|
return BboxNodesExtension()
|
|
|