Add no-box full-frame Qwen AV baselines
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user