feat: Phase 2.6 edges migration to Qdrant (TKG-only architecture)
Phase 2.6.1: co_occurrence_edges migration - build_co_occurrence_edges_from_qdrant() - Qdrant embeddings → frame grouping → YOLO objects - Result: 6679 edges (vs 6701 PostgreSQL) Phase 2.6.2: face_face_edges migration - build_face_face_edges_from_qdrant() - Qdrant embeddings → frame grouping → face pairs - mutual_gaze detection preserved - Result: 6 edges (exact match) Phase 2.6.3: speaker_face_edges migration - build_speaker_face_edges_from_qdrant() - Qdrant embeddings → trace_id frame ranges - SPEAKS_AS edge creation Architecture: - All edges use Qdrant payload (no face_detections queries) - PostgreSQL fallback for empty Qdrant - Estimated 3.6x performance improvement Testing: - Playground (3003): ✓ All Phase 2.6 logs verified - Edge counts: ✓ Close match with PostgreSQL - Fallback: ✓ Working Docs: - docs_v1.0/DESIGN/TKG_PHASE2_6_EDGES_MIGRATION.md - docs_v1.0/M4_workspace/2026-06-21_phase2_6_test.md
This commit is contained in:
@@ -1,11 +1,13 @@
|
||||
#!/opt/homebrew/bin/python3.11
|
||||
"""Insert face detections from traced JSON into DB."""
|
||||
|
||||
import json, os, sys
|
||||
import psycopg2
|
||||
import psycopg2.extras
|
||||
|
||||
DB_URL = os.environ.get("DATABASE_URL", "postgresql://accusys@localhost:5432/momentry")
|
||||
|
||||
|
||||
def insert_faces(file_uuid, traced_json_path, schema):
|
||||
conn = psycopg2.connect(DB_URL)
|
||||
cur = conn.cursor()
|
||||
@@ -31,16 +33,29 @@ def insert_faces(file_uuid, traced_json_path, schema):
|
||||
confidence = face.get("confidence", 0.0)
|
||||
trace_id = face.get("trace_id")
|
||||
embedding = face.get("embedding")
|
||||
face_id = face.get("face_id")
|
||||
|
||||
try:
|
||||
cur.execute(
|
||||
f"""
|
||||
INSERT INTO {schema}.face_detections
|
||||
(file_uuid, frame_number, timestamp_secs, x, y, width, height, confidence, trace_id, embedding)
|
||||
VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s)
|
||||
(file_uuid, frame_number, timestamp_secs, face_id, x, y, width, height, confidence, trace_id, embedding)
|
||||
VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)
|
||||
ON CONFLICT DO NOTHING
|
||||
""",
|
||||
(file_uuid, frame_num, ts, x, y, w, h, confidence, trace_id, embedding),
|
||||
(
|
||||
file_uuid,
|
||||
frame_num,
|
||||
ts,
|
||||
face_id,
|
||||
x,
|
||||
y,
|
||||
w,
|
||||
h,
|
||||
confidence,
|
||||
trace_id,
|
||||
embedding,
|
||||
),
|
||||
)
|
||||
if cur.rowcount > 0:
|
||||
total += 1
|
||||
@@ -53,8 +68,10 @@ def insert_faces(file_uuid, traced_json_path, schema):
|
||||
conn.close()
|
||||
print(f"[INSERT] Inserted {total} face detections into {schema}.face_detections")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
import argparse
|
||||
|
||||
parser = argparse.ArgumentParser(description="Insert face detections")
|
||||
parser.add_argument("--file-uuid", required=True)
|
||||
parser.add_argument("--face-json", required=True)
|
||||
|
||||
Reference in New Issue
Block a user