from .bytetrack import ByteTrack | |
from .dummytrack import DummyTrack | |
def build_tracker(config): | |
""" | |
Build the tracking model based on the provided configuration. | |
""" | |
# Initialize the tracker | |
if config.algorithm == "bytetrack": | |
return ByteTrack() | |
elif config.algorithm == "dummytrack": | |
return DummyTrack() | |
else: | |
raise ValueError(f"Unknown tracker type: {config.tracker.algorithm}") |