PiSAR / pipeline /trackers /__init__.py
eadali's picture
Remove python related files
37165e0
raw
history blame contribute delete
432 Bytes
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}")