This commit is contained in:
2026-03-30 07:22:14 -03:00
parent d0707333fd
commit 4220b0418e
182 changed files with 3668 additions and 5231 deletions

View File

@@ -11,7 +11,7 @@ from __future__ import annotations
import dataclasses
from core.schema.models.pipeline import (
from core.detect.models import (
BoundingBox,
BrandDetection,
BrandStats,
@@ -36,7 +36,7 @@ def serialize_frame_meta(frame: Frame) -> dict:
def serialize_frames_with_upload(frames: list[Frame], job_id: str) -> tuple[list[dict], dict[int, str]]:
"""Upload frame images to S3, return metadata + manifest."""
from detect.checkpoint.frames import save_frames
from core.detect.checkpoint.frames import save_frames
manifest = save_frames(job_id, frames)
meta = [serialize_frame_meta(f) for f in frames]
@@ -45,7 +45,7 @@ def serialize_frames_with_upload(frames: list[Frame], job_id: str) -> tuple[list
def deserialize_frames_with_download(meta: list[dict], manifest: dict, job_id: str) -> list[Frame]:
"""Load frames from S3 + metadata."""
from detect.checkpoint.frames import load_frames
from core.detect.checkpoint.frames import load_frames
int_manifest = {int(k): v for k, v in manifest.items()}
return load_frames(int_manifest, meta)