mariano/mediaproc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

phase 4

Browse Source
This commit is contained in:
buenosairesam
2026-03-30 07:22:14 -03:00
parent d0707333fd
commit 4220b0418e
182 changed files with 3668 additions and 5231 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/detect/manual/run_extract_filter.py
View File

@@ -24,9 +24,9 @@ logging.basicConfig(level=logging.INFO, format="%(levelname)-7s %(name)s — %(m
sys.path.insert(0, ".")
from detect.profiles.soccer import SoccerBroadcastProfile
from detect.stages.frame_extractor import extract_frames
from detect.stages.scene_filter import scene_filter
from core.detect.profile import get_profile
from core.detect.stages.frame_extractor import extract_frames
from core.detect.stages.scene_filter import scene_filter
logger = logging.getLogger(__name__)

4
tests/detect/manual/run_graph.py
View File

@@ -24,8 +24,8 @@ logging.basicConfig(level=logging.INFO, format="%(levelname)-7s %(name)s — %(m
sys.path.insert(0, ".")
from detect.graph import get_pipeline
from detect.state import DetectState
from core.detect.graph import get_pipeline
from core.detect.state import DetectState
logger = logging.getLogger(__name__)

16
tests/detect/manual/run_region_analysis.py
View File

@@ -39,13 +39,13 @@ sys.path.insert(0, ".")
from langgraph.graph import END, StateGraph
from detect import emit
from detect.models import PipelineStats
from detect.profiles.soccer import SoccerBroadcastProfile
from detect.stages.frame_extractor import extract_frames
from detect.stages.scene_filter import scene_filter
from detect.stages.edge_detector import detect_edge_regions
from detect.state import DetectState
from core.detect import emit
from core.detect.models import PipelineStats
from core.detect.profile import get_profile
from core.detect.stages.frame_extractor import extract_frames
from core.detect.stages.scene_filter import scene_filter
from core.detect.stages.edge_detector import detect_edge_regions
from core.detect.state import DetectState
logger = logging.getLogger(__name__)
@@ -166,7 +166,7 @@ def main():
# --- Parameter sensitivity ---
logger.info("=== Parameter sensitivity (local debug) ===")
from detect.stages.edge_detector import _load_cv_edges
from core.detect.stages.edge_detector import _load_cv_edges
edges_mod = _load_cv_edges()
filtered = result.get("filtered_frames", [])

4
tests/detect/manual/seed_scenario.py
View File

@@ -58,7 +58,7 @@ def extract_frames_ffmpeg(video_path: str, fps: float, max_frames: int):
import numpy as np
from PIL import Image
from detect.models import Frame
from core.detect.models import Frame
tmpdir = tempfile.mkdtemp(prefix="scenario_")
pattern = os.path.join(tmpdir, "frame_%04d.jpg")
@@ -111,7 +111,7 @@ def main():
logger.info("Extracted %d frames", len(frames))
# Create timeline + branch + checkpoint
from detect.checkpoint.storage import create_timeline, save_stage_output
from core.detect.checkpoint.storage import create_timeline, save_stage_output
timeline_id, branch_id = create_timeline(
source_video=video_path,

2
tests/detect/manual/test_cloud_provider.py
View File

@@ -58,7 +58,7 @@ def make_brand_image(text: str, width: int = 300, height: int = 100) -> str:
def main():
from detect.providers import get_provider, has_api_key, PROVIDERS
from core.detect.providers import get_provider, has_api_key, PROVIDERS
provider_name = os.environ.get("CLOUD_LLM_PROVIDER", "groq")
logger.info("Provider: %s", provider_name)

4
tests/detect/manual/test_frame_extractor_e2e.py
View File

@@ -13,8 +13,8 @@ import sys
sys.path.insert(0, ".")
from detect.profiles.soccer import SoccerBroadcastProfile
from detect.stages.frame_extractor import extract_frames
from core.detect.profile import get_profile
from core.detect.stages.frame_extractor import extract_frames
logger = logging.getLogger(__name__)

6
tests/detect/manual/test_ocr_e2e.py
View File

@@ -86,9 +86,9 @@ def test_ocr_stage_remote(url: str):
logger.info("--- OCR stage (remote mode) ---")
sys.path.insert(0, ".")
from detect.models import BoundingBox, Frame
from detect.profiles.base import OCRConfig
from detect.stages.ocr_stage import run_ocr
from core.detect.models import BoundingBox, Frame
from core.detect.stages.models import OCRConfig
from core.detect.stages.ocr_stage import run_ocr
# Create a frame with text baked in
image = make_text_image("EMIRATES")

8
tests/detect/manual/test_replay.py
View File

@@ -48,10 +48,10 @@ def main():
# Override Redis to localhost (ctrl/.env has k8s hostname)
os.environ["REDIS_URL"] = f"redis://localhost:{args.port}/0"
from detect.graph import get_pipeline, NODES
from detect.checkpoint import list_checkpoints
from detect.checkpoint import replay_from
from detect.state import DetectState
from core.detect.graph import get_pipeline, NODES
from core.detect.checkpoint import list_checkpoints
from core.detect.checkpoint import replay_from
from core.detect.state import DetectState
VIDEO = "media/out/chunks/95043d50-4df6-4ac8-bbd5-2ba873117c6e/chunk_0000.mp4"