25 lines
650 B
Python
25 lines
650 B
Python
"""
|
|
ComfyUI-Grounding: Simplified object detection nodes for ComfyUI
|
|
Supports GroundingDINO, MM-GroundingDINO, OWLv2, Florence-2, and YOLO-World
|
|
"""
|
|
|
|
from .nodes import (
|
|
GroundingModelLoader,
|
|
GroundingDetector,
|
|
BboxVisualizer,
|
|
)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"GroundingModelLoader": GroundingModelLoader,
|
|
"GroundingDetector": GroundingDetector,
|
|
"BboxVisualizer": BboxVisualizer,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"GroundingModelLoader": "Grounding Model Loader",
|
|
"GroundingDetector": "Grounding Detector",
|
|
"BboxVisualizer": "Bounding Box Visualizer",
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|