Add causal person fusion and MuVAP stage-3 baselines

This commit is contained in:
Xin Wang
2026-09-10 06:54:22 +08:00
parent 0534c9d3d0
commit d426b4740a
225 changed files with 27716 additions and 119 deletions
+26
View File
@@ -0,0 +1,26 @@
# dimension of features extracted from different models
# used in feature_head, dataset, models
FEATURE_SIZES = {
'rgb_anet_resnet50': 2048,
'flow_anet_resnet50': 2048,
'rgb_kinetics_bninception': 1024,
'flow_kinetics_bninception': 1024,
'rgb_farl': 512,
'rgb_kinetics_resnet50': 2048,
'rgb_kinetics_resnet50_debug': 2048,
'rgb_kinetics_resnet50_debug_resized': 2048,
'flow_kinetics_resnet50': 2048,
'flow_nv_kinetics_bninception': 1024,
'rgb_kinetics_i3d': 2048,
'flow_kinetics_i3d': 2048,
'audio_wav2vec2': 5120,
'audio_wav2vec2_left': 5120,
'audio_wav2vec2_right': 5120,
'text_word2vec': 300,
'text_fasttext_crawl': 300,
'text_gpt2_last_subword_window_size_0': 768,
'text_gpt2_last_subword_window_size_20': 768,
'text_gpt2_last_subword_window_size_50': 768,
'text_gpt2_last_subword_window_size_100': 768,
}