Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion apps/backend/dependencies.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ async def get_redis_client() -> aioredis.Redis:
"""Get or create Redis client from REDIS_URL env var."""
global _redis_client
if _redis_client is None:
redis_url = os.getenv("REDIS_URL", settings.REDIS_URL)
redis_url = os.getenv("REDIS_URL", settings.redis_url)
_redis_client = await aioredis.from_url(redis_url, decode_responses=True)
return _redis_client

Expand Down
4 changes: 2 additions & 2 deletions benchmark.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
from ultralytics import YOLO # <-- FIX: Loaded successfully

class PipelineBenchmark:
def __init__(self, redis_url=settings.REDIS_URL):
def __init__(self, redis_url=settings.redis_url):
self.redis_url = redis_url
self.metrics = {
"detection_times": [],
Expand Down Expand Up @@ -314,7 +314,7 @@ def run_comparative_benchmark(benchrunner, models, num_frames=100):
parser.add_argument("--frames", type=int, default=100, help="Number of frames to benchmark")
args = parser.parse_args()

REDIS_ENV_URL = os.getenv("REDIS_URL", settings.REDIS_URL)
REDIS_ENV_URL = os.getenv("REDIS_URL", settings.redis_url)
benchrunner = PipelineBenchmark(redis_url=REDIS_ENV_URL)

if args.model:
Expand Down
2 changes: 1 addition & 1 deletion scripts/export_finetuning_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,7 @@ def parse_args():

parser.add_argument(
"--redis-url",
default=os.getenv("REDIS_URL", settings.REDIS_URL),
default=os.getenv("REDIS_URL", settings.redis_url),
help="Redis connection URL (default: env REDIS_URL or localhost:6379)"
)
parser.add_argument(
Expand Down
2 changes: 1 addition & 1 deletion scripts/inspect_tracks.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
import redis

DEFAULT_CAMERA_ID = "cam_01"
DEFAULT_REDIS_URL = settings.REDIS_URL
DEFAULT_REDIS_URL = settings.redis_url


@dataclass
Expand Down
12 changes: 1 addition & 11 deletions services/tracking/tracker.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,16 +72,6 @@ def __init__(
reid_similarity_threshold: float = 0.85,
max_interpolation_gap: int = 10, # Added with a sensible default
) -> None:
"""Initialize the Tracker with DeepSort backend and internal state.

Args:
fps: Frames per second of the input stream.
max_age: Frames before a lost track is marked DEAD.
n_init: Frames before a track is CONFIRMED.
max_cosine_distance: ReID appearance distance threshold.
camera_id: Identifier for the camera feed.
event_logger: Optional logger for lifecycle events.
reid_similarity_threshold: Cosine similarity cutoff for ReID matching.
"""Initialize the tracker with DeepSort hyperparameters and interpolation constraints.

Args:
Expand Down Expand Up @@ -139,7 +129,7 @@ def update(

Args:
det_frame: Output of Phase 1 detector (DetectionFrameSchema).
raw_frame: Original BGR frame needed for appearance features.
raw_frame: Original BGR frame - needed for appearance features.

Returns:
A ``TrackedFrame`` containing all confirmed tracks for this frame,
Expand Down
4 changes: 2 additions & 2 deletions tests/test_inspect_tracks.py
Original file line number Diff line number Diff line change
Expand Up @@ -170,7 +170,7 @@ def test_json_payload_is_machine_readable(fake_redis):

encoded = json.dumps(payload)
decoded = json.loads(encoded)
assert decoded["redis_url"] == settings.REDIS_URL
assert decoded["redis_url"] == settings.redis_url
assert "secret" not in encoded
assert decoded["tracks"][0]["track_id"] == 3
assert decoded["tracks"][0]["events"][0]["event"] == "BORN"
Expand All @@ -184,7 +184,7 @@ def test_text_render_includes_detail_rows_for_single_track(fake_redis):
output = render_text(
summaries,
camera_id="cam_01",
redis_url=settings.REDIS_URL,
redis_url=settings.redis_url,
show_event_rows=True,
)

Expand Down