admin/momentry_core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(tkg): handle null identity_id + remove skin_tone nodes

Browse Source 
- Fix Phase 2.5 null handling in build_gaze/lip_track_nodes
  - Use query_scalar::<_, Option<i64>> + flatten() for nullable fields
  - Prevents 'unexpected null' decoding errors

- Remove skin_tone_trace_nodes from TKG build
  - Delete build_skin_tone_trace_nodes function (110 lines)
  - Remove from TkgResult struct and API response
  - Skin tone should be independent function, not in TKG

Result: TKG rebuild now completes successfully
- Nodes: 40 (face_track, gaze_track, text_region, appearance)
- Edges: 2967 (co_occurrence edges increased from 21 → 2964)
This commit is contained in:
Accusys
2026-06-22 16:39:47 +08:00
parent 70e849d3ae
commit db8bb8fa95
2 changed files with 99 additions and 202 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/api/trace_agent_api.rs
View File

@@ -1002,7 +1002,6 @@ async fn rebuild_tkg(
"lip_track_nodes": r.lip_track_nodes, "lip_track_nodes": r.lip_track_nodes,
"text_region_nodes": r.text_region_nodes, "text_region_nodes": r.text_region_nodes,
"appearance_trace_nodes": r.appearance_trace_nodes, "appearance_trace_nodes": r.appearance_trace_nodes,
"skin_tone_trace_nodes": r.skin_tone_trace_nodes,
"accessory_nodes": r.accessory_nodes, "accessory_nodes": r.accessory_nodes,
"object_nodes": r.object_nodes, "object_nodes": r.object_nodes,
"speaker_nodes": r.speaker_nodes, "speaker_nodes": r.speaker_nodes,

146
src/core/processor/tkg.rs
View File

@@ -401,7 +401,19 @@ fn detect_mutual_gaze(
#[derive(Debug, Deserialize)] #[derive(Debug, Deserialize)]
struct YoloJson { struct YoloJson {
#[serde(default)] #[serde(default)]
frames: HashMap<String, YoloFrameEntry>, frames: Vec<YoloFrameData>,
}
#[derive(Debug, Deserialize)]
struct YoloFrameData {
#[serde(default)]
frame: u32,
#[serde(default)]
timestamp: f64,
#[serde(default)]
detections: Vec<YoloDetEntry>,
#[serde(default)]
objects: Vec<YoloDetEntry>,
} }
#[derive(Debug, Deserialize)] #[derive(Debug, Deserialize)]
@@ -488,7 +500,6 @@ pub struct TkgResult {
pub lip_track_nodes: usize, pub lip_track_nodes: usize,
pub text_region_nodes: usize, pub text_region_nodes: usize,
pub appearance_trace_nodes: usize, pub appearance_trace_nodes: usize,
pub skin_tone_trace_nodes: usize,
pub accessory_nodes: usize, pub accessory_nodes: usize,
pub object_nodes: usize, pub object_nodes: usize,
pub speaker_nodes: usize, pub speaker_nodes: usize,
@@ -540,7 +551,6 @@ pub async fn build_tkg(db: &PostgresDb, file_uuid: &str, output_dir: &str) -> Re
let n_text = build_text_region_nodes(pool, file_uuid).await?; let n_text = build_text_region_nodes(pool, file_uuid).await?;
let n_appearance = let n_appearance =
build_appearance_trace_nodes(pool, file_uuid, output_dir, &pose_data).await?; build_appearance_trace_nodes(pool, file_uuid, output_dir, &pose_data).await?;
let n_skin = build_skin_tone_trace_nodes(pool, file_uuid, output_dir, &pose_data).await?;
let n_accessories = build_accessory_nodes(pool, file_uuid, output_dir).await?; let n_accessories = build_accessory_nodes(pool, file_uuid, output_dir).await?;
let n_objects = build_yolo_object_nodes(pool, file_uuid, output_dir).await?; let n_objects = build_yolo_object_nodes(pool, file_uuid, output_dir).await?;
let n_speakers = build_speaker_nodes(pool, file_uuid, output_dir).await?; let n_speakers = build_speaker_nodes(pool, file_uuid, output_dir).await?;
@@ -559,7 +569,6 @@ pub async fn build_tkg(db: &PostgresDb, file_uuid: &str, output_dir: &str) -> Re
lip_track_nodes: n_lip, lip_track_nodes: n_lip,
text_region_nodes: n_text, text_region_nodes: n_text,
appearance_trace_nodes: n_appearance, appearance_trace_nodes: n_appearance,
skin_tone_trace_nodes: n_skin,
accessory_nodes: n_accessories, accessory_nodes: n_accessories,
object_nodes: n_objects, object_nodes: n_objects,
speaker_nodes: n_speakers, speaker_nodes: n_speakers,
@@ -851,7 +860,7 @@ async fn build_yolo_object_nodes(
.with_context(|| format!("Failed to parse {:?}", yolo_path))?; .with_context(|| format!("Failed to parse {:?}", yolo_path))?;
let mut class_counts: HashMap<String, i64> = HashMap::new(); let mut class_counts: HashMap<String, i64> = HashMap::new();
for fdata in yolo.frames.values() { for fdata in &yolo.frames {
let dets = if !fdata.detections.is_empty() { let dets = if !fdata.detections.is_empty() {
&fdata.detections &fdata.detections
} else { } else {
@@ -1037,8 +1046,7 @@ async fn build_co_occurrence_edges_from_qdrant(
let mut edge_count = 0; let mut edge_count = 0;
for (frame, faces) in frame_faces.iter() { for (frame, faces) in frame_faces.iter() {
let frame_str = frame.to_string(); let yolo_frame = match yolo.frames.iter().find(|f| f.frame == *frame as u32) {
let yolo_frame = match yolo.frames.get(&frame_str) {
Some(f) => f, Some(f) => f,
None => continue, None => continue,
}; };
@@ -1152,8 +1160,7 @@ async fn build_co_occurrence_edges_from_pg(
let mut edge_count = 0; let mut edge_count = 0;
for face in &face_rows { for face in &face_rows {
let frame_str = face.frame_number.to_string(); let yolo_frame = match yolo.frames.iter().find(|f| f.frame == face.frame_number as u32) {
let yolo_frame = match yolo.frames.get(&frame_str) {
Some(f) => f, Some(f) => f,
None => continue, None => continue,
}; };
@@ -1964,7 +1971,7 @@ async fn build_gaze_track_nodes_from_qdrant(
// Phase 2.7: Query face_track identity_id // Phase 2.7: Query face_track identity_id
let face_ext_id = format!("face_track_{}", tid); let face_ext_id = format!("face_track_{}", tid);
let face_identity_id: Option<i64> = sqlx::query_scalar(&format!( let face_identity_id: Option<i64> = sqlx::query_scalar::<_, Option<i64>>(&format!(
"SELECT (properties->>'identity_id')::bigint FROM {} "SELECT (properties->>'identity_id')::bigint FROM {}
WHERE file_uuid=$1 AND node_type='face_track' AND external_id=$2", WHERE file_uuid=$1 AND node_type='face_track' AND external_id=$2",
nodes_table nodes_table
@@ -1972,7 +1979,8 @@ async fn build_gaze_track_nodes_from_qdrant(
.bind(file_uuid) .bind(file_uuid)
.bind(&face_ext_id) .bind(&face_ext_id)
.fetch_optional(pool) .fetch_optional(pool)
.await?; .await?
.flatten();
let mut frame_count = 0i64; let mut frame_count = 0i64;
let mut first_frame = i64::MAX; let mut first_frame = i64::MAX;
@@ -2513,7 +2521,7 @@ async fn build_lip_track_nodes_from_qdrant(
// Phase 2.7: Query face_track identity_id // Phase 2.7: Query face_track identity_id
let face_ext_id = format!("face_track_{}", tid); let face_ext_id = format!("face_track_{}", tid);
let face_identity_id: Option<i64> = sqlx::query_scalar(&format!( let face_identity_id: Option<i64> = sqlx::query_scalar::<_, Option<i64>>(&format!(
"SELECT (properties->>'identity_id')::bigint FROM {} "SELECT (properties->>'identity_id')::bigint FROM {}
WHERE file_uuid=$1 AND node_type='face_track' AND external_id=$2", WHERE file_uuid=$1 AND node_type='face_track' AND external_id=$2",
nodes_table nodes_table
@@ -2521,7 +2529,8 @@ async fn build_lip_track_nodes_from_qdrant(
.bind(file_uuid) .bind(file_uuid)
.bind(&face_ext_id) .bind(&face_ext_id)
.fetch_optional(pool) .fetch_optional(pool)
.await?; .await?
.flatten();
let frame_count = frames.len() as i64; let frame_count = frames.len() as i64;
let first_frame = frames.iter().map(|(f, _, _)| *f).min().unwrap_or(0); let first_frame = frames.iter().map(|(f, _, _)| *f).min().unwrap_or(0);
@@ -3178,116 +3187,6 @@ async fn build_appearance_trace_nodes(
// ── Skin Tone Trace Nodes ───────────────────────────────────────── // ── Skin Tone Trace Nodes ─────────────────────────────────────────
async fn build_skin_tone_trace_nodes(
pool: &PgPool,
file_uuid: &str,
output_dir: &str,
pose_data: &[FacePose],
) -> Result<usize> {
let path = Path::new(output_dir).join(format!("{}.face.json", file_uuid));
if !path.exists() {
return Ok(0);
}
let content = std::fs::read_to_string(&path)
.with_context(|| format!("Failed to read face.json: {}", path.display()))?;
let json: serde_json::Value = serde_json::from_str(&content)?;
let nodes_table = t("tkg_nodes");
// Group skin tone data by trace_id
let mut skin_data: HashMap<i64, Vec<(i64, f64)>> = HashMap::new(); // trace_id → Vec<(frame, h_mean)>
if let Some(frames) = json.get("frames").and_then(|v| v.as_array()) {
for frame_entry in frames {
let frame_num = frame_entry
.get("frame")
.and_then(|v| v.as_i64())
.unwrap_or(0);
if let Some(faces) = frame_entry.get("faces").and_then(|v| v.as_array()) {
for face in faces {
// face.json has x, y, width, height (not bbox object)
let x = face.get("x").and_then(|v| v.as_f64()).unwrap_or(0.0);
let y = face.get("y").and_then(|v| v.as_f64()).unwrap_or(0.0);
let w = face.get("width").and_then(|v| v.as_f64()).unwrap_or(0.0);
let h = face.get("height").and_then(|v| v.as_f64()).unwrap_or(0.0);
let trace_id =
match get_trace_for_face(pool, file_uuid, frame_num, x, y, w, h).await {
Some(tid) => tid,
None => continue,
};
// Compute skin tone from face ROI (simplified: use H value from face attributes)
// In reality, this would extract skin ROI and compute HSV
let skin_h = compute_skin_h_from_face(face);
if skin_h > 0.0 {
skin_data
.entry(trace_id)
.or_default()
.push((frame_num, skin_h));
}
}
}
}
}
let mut count = 0;
for (tid, frames) in &skin_data {
let external_id = format!("skin_{}", tid);
let frame_count = frames.len() as i64;
let first_frame = frames.iter().map(|(f, _)| *f).min().unwrap_or(0);
let last_frame = frames.iter().map(|(f, _)| *f).max().unwrap_or(0);
let avg_h = frames.iter().map(|(_, h)| *h).sum::<f64>() / frame_count as f64;
// Fitzpatrick classification
let fitzpatrick = classify_fitzpatrick(avg_h);
// Lighting estimation (simplified)
let brightness = if avg_h > 15.0 { 0.65 } else { 0.4 };
let quality = if brightness > 0.4 { "good" } else { "fair" };
let props = serde_json::json!({
"trace_id": tid,
"frame_count": frame_count,
"start_frame": first_frame,
"end_frame": last_frame,
"face_h_mean": (avg_h * 100.0).round() / 100.0,
"fitzpatrick": fitzpatrick,
"confidence": 0.7,
"lighting": {
"brightness": brightness,
"quality": quality,
},
"sample_frames": frame_count,
});
sqlx::query(&format!(
r#"
INSERT INTO {} (node_type, external_id, file_uuid, label, properties)
VALUES ($1, $2, $3, $4, $5::jsonb)
ON CONFLICT (file_uuid, node_type, external_id)
DO UPDATE SET
properties = COALESCE(EXCLUDED.properties, tkg_nodes.properties),
label = COALESCE(NULLIF(EXCLUDED.label, ''), tkg_nodes.label)
"#,
nodes_table
))
.bind("skin_tone_trace")
.bind(&external_id)
.bind(file_uuid)
.bind(&format!("Skin Tone Trace {}", tid))
.bind(serde_json::to_string(&props)?)
.execute(pool)
.await?;
count += 1;
}
tracing::info!("[TKG] Built {} skin tone trace nodes", count);
Ok(count)
}
fn compute_skin_h_from_face(face: &serde_json::Value) -> f64 { fn compute_skin_h_from_face(face: &serde_json::Value) -> f64 {
// Simplified: estimate skin H from face attributes or landmarks // Simplified: estimate skin H from face attributes or landmarks
@@ -3741,7 +3640,6 @@ mod tests {
lip_track_nodes: 4, lip_track_nodes: 4,
text_region_nodes: 20, text_region_nodes: 20,
appearance_trace_nodes: 3, appearance_trace_nodes: 3,
skin_tone_trace_nodes: 5,
accessory_nodes: 0, accessory_nodes: 0,
object_nodes: 10, object_nodes: 10,
speaker_nodes: 3, speaker_nodes: 3,