Add no-box full-frame Qwen AV baselines

This commit is contained in:
Xin Wang
2026-09-17 02:51:26 +08:00
parent a9f748d6a6
commit 62134a0fa0
10 changed files with 1164 additions and 17 deletions

View File

@@ -14,12 +14,12 @@ from tqdm.auto import tqdm
from easycom_head import (
CLASS_NAMES,
DATASET_NAMES,
HISTORY_VISUAL_SET_FUSION_MODES,
LOCAL_FUSION_SCOPES,
NUM_CLASSES,
NUM_HORIZONS,
OFFICIAL_TEST_SESSIONS,
STEP_SECONDS,
TRACK_PERSON_FUSION_MODES,
TurnPredictionHead,
average_precision_metrics,
checkpoint_head_config,
@@ -150,7 +150,7 @@ def _predict_states(
start,
min(start + batch_size, len(states)),
)
if head.local_fusion in TRACK_PERSON_FUSION_MODES:
if head.local_fusion in HISTORY_VISUAL_SET_FUSION_MODES:
person_batch, person_mask = local_person_features.padded_history(
rows,
max_observations=head.person_history_frames,