├── .gitignore ├── Evaluation ├── README.md ├── eval_detection.py └── utils.py ├── LICENSE ├── README.md ├── configs ├── thumos14_e2e_slowfast_tadtr.yml └── thumos14_i3d2s_tadtr.yml ├── datasets ├── __init__.py ├── data_utils.py ├── e2e_lib │ ├── __init__.py │ ├── image_utils.py │ └── videotransforms.py ├── path.yml ├── tad_dataset.py └── tad_eval.py ├── demo.py ├── docs └── 1_train_on_your_dataset.md ├── engine.py ├── figs └── speed-accuracy.png ├── main.py ├── models ├── __init__.py ├── custom_loss.py ├── matcher.py ├── ops │ ├── roi_align │ │ ├── __init__.py │ │ ├── roi_align.py │ │ └── src │ │ │ ├── roi_align_cuda.cpp │ │ │ └── roi_align_kernel.cu │ ├── setup.py │ └── temporal_deform_attn │ │ ├── __init__.py │ │ └── temporal_deform_attn.py ├── position_encoding.py ├── tadtr.py ├── transformer.py ├── video_encoder.py └── video_encoder_archs │ ├── __init__.py │ ├── resnet3d.py │ ├── slowfast.py │ └── tsm.py ├── opts.py ├── requirements.txt ├── scripts ├── run_parallel.sh ├── test_reference_models.sh └── test_reference_models_e2e.sh ├── tools ├── __init__.py ├── extract_frames.py ├── flop_count.py ├── prepare_data.py └── test_runtime.py └── util ├── __init__.py ├── logger.py ├── misc.py └── segment_ops.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/.gitignore -------------------------------------------------------------------------------- /Evaluation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/Evaluation/README.md -------------------------------------------------------------------------------- /Evaluation/eval_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/Evaluation/eval_detection.py -------------------------------------------------------------------------------- /Evaluation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/Evaluation/utils.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/README.md -------------------------------------------------------------------------------- /configs/thumos14_e2e_slowfast_tadtr.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/configs/thumos14_e2e_slowfast_tadtr.yml -------------------------------------------------------------------------------- /configs/thumos14_i3d2s_tadtr.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/configs/thumos14_i3d2s_tadtr.yml -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/datasets/data_utils.py -------------------------------------------------------------------------------- /datasets/e2e_lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/datasets/e2e_lib/__init__.py -------------------------------------------------------------------------------- /datasets/e2e_lib/image_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/datasets/e2e_lib/image_utils.py -------------------------------------------------------------------------------- /datasets/e2e_lib/videotransforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/datasets/e2e_lib/videotransforms.py -------------------------------------------------------------------------------- /datasets/path.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/datasets/path.yml -------------------------------------------------------------------------------- /datasets/tad_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/datasets/tad_dataset.py -------------------------------------------------------------------------------- /datasets/tad_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/datasets/tad_eval.py -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/demo.py -------------------------------------------------------------------------------- /docs/1_train_on_your_dataset.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/docs/1_train_on_your_dataset.md -------------------------------------------------------------------------------- /engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/engine.py -------------------------------------------------------------------------------- /figs/speed-accuracy.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/figs/speed-accuracy.png -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/custom_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/models/custom_loss.py -------------------------------------------------------------------------------- /models/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/models/matcher.py -------------------------------------------------------------------------------- /models/ops/roi_align/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/models/ops/roi_align/__init__.py -------------------------------------------------------------------------------- /models/ops/roi_align/roi_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/models/ops/roi_align/roi_align.py -------------------------------------------------------------------------------- /models/ops/roi_align/src/roi_align_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/models/ops/roi_align/src/roi_align_cuda.cpp -------------------------------------------------------------------------------- /models/ops/roi_align/src/roi_align_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/models/ops/roi_align/src/roi_align_kernel.cu -------------------------------------------------------------------------------- /models/ops/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/models/ops/setup.py -------------------------------------------------------------------------------- /models/ops/temporal_deform_attn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/models/ops/temporal_deform_attn/__init__.py -------------------------------------------------------------------------------- /models/ops/temporal_deform_attn/temporal_deform_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/models/ops/temporal_deform_attn/temporal_deform_attn.py -------------------------------------------------------------------------------- /models/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/models/position_encoding.py -------------------------------------------------------------------------------- /models/tadtr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/models/tadtr.py -------------------------------------------------------------------------------- /models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/models/transformer.py -------------------------------------------------------------------------------- /models/video_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/models/video_encoder.py -------------------------------------------------------------------------------- /models/video_encoder_archs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/video_encoder_archs/resnet3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/models/video_encoder_archs/resnet3d.py -------------------------------------------------------------------------------- /models/video_encoder_archs/slowfast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/models/video_encoder_archs/slowfast.py -------------------------------------------------------------------------------- /models/video_encoder_archs/tsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/models/video_encoder_archs/tsm.py -------------------------------------------------------------------------------- /opts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/opts.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/run_parallel.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/scripts/run_parallel.sh -------------------------------------------------------------------------------- /scripts/test_reference_models.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/scripts/test_reference_models.sh -------------------------------------------------------------------------------- /scripts/test_reference_models_e2e.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/scripts/test_reference_models_e2e.sh -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/extract_frames.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/tools/extract_frames.py -------------------------------------------------------------------------------- /tools/flop_count.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/tools/flop_count.py -------------------------------------------------------------------------------- /tools/prepare_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/tools/prepare_data.py -------------------------------------------------------------------------------- /tools/test_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/tools/test_runtime.py -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/util/__init__.py -------------------------------------------------------------------------------- /util/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/util/logger.py -------------------------------------------------------------------------------- /util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/util/misc.py -------------------------------------------------------------------------------- /util/segment_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlliu7/E2E-TAD/HEAD/util/segment_ops.py --------------------------------------------------------------------------------