13 lines
372 B
Python
13 lines
372 B
Python
from .nodes.yolo_tracker_node import YOLOTrackerNode
|
|
from .nodes.openpose_tracker_node import OpenPoseTrackerNode
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"YOLOTrackerNode": YOLOTrackerNode,
|
|
"OpenPoseTrackerNode": OpenPoseTrackerNode,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"YOLOTrackerNode": "YOLO BoundingBox Tracker",
|
|
"OpenPoseTrackerNode": "OpenPose BoundingBox Tracker"
|
|
}
|