├── .gitignore ├── INSTALL.md ├── README.md ├── eval.py ├── libs ├── __init__.py ├── core │ ├── __init__.py │ └── opt.py ├── data │ ├── __init__.py │ ├── data_utils.py │ ├── dataset.py │ └── tokenizer.py ├── dist_utils.py ├── modeling │ ├── __init__.py │ ├── blocks.py │ ├── fusion.py │ ├── head.py │ ├── loss.py │ ├── model.py │ ├── optim.py │ ├── text_net.py │ ├── video_net.py │ └── weight_init.py ├── nms │ ├── __init__.py │ ├── nms.py │ ├── setup_nms.py │ └── src │ │ └── nms_cpu.cpp ├── train_utils.py └── worker.py ├── opts └── video_centric │ ├── anet_1.3.yaml │ ├── charades_sta_c3d.yaml │ ├── charades_sta_i3d.yaml │ ├── ego4d_egovlp.yaml │ ├── ego4d_slowfast_bert.yaml │ ├── mad.yaml │ └── tacos.yaml └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/.gitignore -------------------------------------------------------------------------------- /INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/INSTALL.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/README.md -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/eval.py -------------------------------------------------------------------------------- /libs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/__init__.py -------------------------------------------------------------------------------- /libs/core/__init__.py: -------------------------------------------------------------------------------- 1 | from .opt import load_opt -------------------------------------------------------------------------------- /libs/core/opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/core/opt.py -------------------------------------------------------------------------------- /libs/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/data/__init__.py -------------------------------------------------------------------------------- /libs/data/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/data/data_utils.py -------------------------------------------------------------------------------- /libs/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/data/dataset.py -------------------------------------------------------------------------------- /libs/data/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/data/tokenizer.py -------------------------------------------------------------------------------- /libs/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/dist_utils.py -------------------------------------------------------------------------------- /libs/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/modeling/__init__.py -------------------------------------------------------------------------------- /libs/modeling/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/modeling/blocks.py -------------------------------------------------------------------------------- /libs/modeling/fusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/modeling/fusion.py -------------------------------------------------------------------------------- /libs/modeling/head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/modeling/head.py -------------------------------------------------------------------------------- /libs/modeling/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/modeling/loss.py -------------------------------------------------------------------------------- /libs/modeling/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/modeling/model.py -------------------------------------------------------------------------------- /libs/modeling/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/modeling/optim.py -------------------------------------------------------------------------------- /libs/modeling/text_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/modeling/text_net.py -------------------------------------------------------------------------------- /libs/modeling/video_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/modeling/video_net.py -------------------------------------------------------------------------------- /libs/modeling/weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/modeling/weight_init.py -------------------------------------------------------------------------------- /libs/nms/__init__.py: -------------------------------------------------------------------------------- 1 | from .nms import batched_nms -------------------------------------------------------------------------------- /libs/nms/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/nms/nms.py -------------------------------------------------------------------------------- /libs/nms/setup_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/nms/setup_nms.py -------------------------------------------------------------------------------- /libs/nms/src/nms_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/nms/src/nms_cpu.cpp -------------------------------------------------------------------------------- /libs/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/train_utils.py -------------------------------------------------------------------------------- /libs/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/libs/worker.py -------------------------------------------------------------------------------- /opts/video_centric/anet_1.3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/opts/video_centric/anet_1.3.yaml -------------------------------------------------------------------------------- /opts/video_centric/charades_sta_c3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/opts/video_centric/charades_sta_c3d.yaml -------------------------------------------------------------------------------- /opts/video_centric/charades_sta_i3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/opts/video_centric/charades_sta_i3d.yaml -------------------------------------------------------------------------------- /opts/video_centric/ego4d_egovlp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/opts/video_centric/ego4d_egovlp.yaml -------------------------------------------------------------------------------- /opts/video_centric/ego4d_slowfast_bert.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/opts/video_centric/ego4d_slowfast_bert.yaml -------------------------------------------------------------------------------- /opts/video_centric/mad.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/opts/video_centric/mad.yaml -------------------------------------------------------------------------------- /opts/video_centric/tacos.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/opts/video_centric/tacos.yaml -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fmu2/snag_release/HEAD/train.py --------------------------------------------------------------------------------